ellawang9 commited on
Commit
d94a89d
·
verified ·
1 Parent(s): 7a181ec

Update config.py

Browse files
Files changed (1) hide show
  1. config.py +4 -3
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
- # 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
 
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")