sdufays commited on
Commit
98c934e
·
1 Parent(s): 6d26057

Update config.py

Browse files
Files changed (1) hide show
  1. config.py +2 -1
config.py CHANGED
@@ -5,8 +5,9 @@ from dotenv import load_dotenv
5
  load_dotenv()
6
 
7
 
8
- BASE_MODEL = "TinyLlama/TinyLlama-1.1B-Chat-v1.0"
9
  # Other options:
 
10
  # MODEL = "meta-llama/Llama-2-7b-chat-hf"
11
  # MODEL = "openlm-research/open_llama_3b"
12
 
 
5
  load_dotenv()
6
 
7
 
8
+ BASE_MODEL = "mistralai/Mistral-7B-Instruct-v0.1" # just testing
9
  # Other options:
10
+ # MODEL = "TinyLlama/TinyLlama-1.1B-Chat-v1.0"" --> original recommended model
11
  # MODEL = "meta-llama/Llama-2-7b-chat-hf"
12
  # MODEL = "openlm-research/open_llama_3b"
13