Spaces:
Build error
Build error
Update config.py
Browse files
config.py
CHANGED
|
@@ -5,8 +5,9 @@ from dotenv import load_dotenv
|
|
| 5 |
load_dotenv()
|
| 6 |
|
| 7 |
|
| 8 |
-
BASE_MODEL = "
|
| 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 |
|