Spaces:
Sleeping
Sleeping
Update config.py
Browse files
config.py
CHANGED
@@ -5,13 +5,14 @@ from dotenv import load_dotenv
|
|
5 |
load_dotenv()
|
6 |
|
7 |
# BASE_MODEL = "HuggingFaceH4/zephyr-7b-beta"
|
8 |
-
BASE_MODEL = "TinyLlama/TinyLlama-1.1B-Chat-v1.0"
|
9 |
-
|
10 |
# Other options:
|
11 |
# MODEL = "meta-llama/Llama-2-7b-chat-hf"
|
12 |
# MODEL = "openlm-research/open_llama_3b"
|
13 |
|
14 |
# If you finetune the model or change it in any way, save it to huggingface hub, then set MY_MODEL to your model ID. The model ID is in the format "your-username/your-model-name".
|
15 |
-
MY_MODEL = "ellawang9/bps-chatbot"
|
|
|
16 |
|
17 |
HF_TOKEN = os.getenv("HF_TOKEN")
|
|
|
5 |
load_dotenv()
|
6 |
|
7 |
# BASE_MODEL = "HuggingFaceH4/zephyr-7b-beta"
|
8 |
+
# BASE_MODEL = "TinyLlama/TinyLlama-1.1B-Chat-v1.0"
|
9 |
+
BASE_MODEL = "Qwen/Qwen2.5-3B"
|
10 |
# Other options:
|
11 |
# MODEL = "meta-llama/Llama-2-7b-chat-hf"
|
12 |
# MODEL = "openlm-research/open_llama_3b"
|
13 |
|
14 |
# If you finetune the model or change it in any way, save it to huggingface hub, then set MY_MODEL to your model ID. The model ID is in the format "your-username/your-model-name".
|
15 |
+
# MY_MODEL = "ellawang9/bps-chatbot"
|
16 |
+
MY_MODEL = None
|
17 |
|
18 |
HF_TOKEN = os.getenv("HF_TOKEN")
|