arya-ai-model commited on
Commit
88267dd
·
1 Parent(s): 355b8bb

First commit

Browse files
Files changed (1) hide show
  1. app/config.py +1 -1
app/config.py CHANGED
@@ -1,5 +1,5 @@
1
  import os
2
 
3
- MODEL_NAME = "deepseek-ai/deepseek-llm-r1-7b"
4
  HF_TOKEN = os.getenv("HF_TOKEN") # Set this in the Hugging Face environment
5
  DEVICE = "cuda" # Change to "cpu" if no GPU
 
1
  import os
2
 
3
+ MODEL_NAME = "deepseek-ai/deepseek-llm-7b-base"
4
  HF_TOKEN = os.getenv("HF_TOKEN") # Set this in the Hugging Face environment
5
  DEVICE = "cuda" # Change to "cpu" if no GPU