wangyoucao577 commited on
Commit
9408e2e
·
verified ·
1 Parent(s): ba8572c

feat: new model

Browse files
Files changed (1) hide show
  1. entrypoint.sh +2 -1
entrypoint.sh CHANGED
@@ -2,7 +2,8 @@
2
 
3
  mkdir -p data
4
 
5
- HF_MODEL_PATH=Qwen/Qwen2.5-1.5B-Instruct
 
6
  HF_MODEL_NAME=$(basename ${HF_MODEL_PATH})
7
  LOCAL_MODEL_PATH=./data/${HF_MODEL_NAME}
8
 
 
2
 
3
  mkdir -p data
4
 
5
+ #HF_MODEL_PATH=Qwen/Qwen2.5-1.5B-Instruct
6
+ HF_MODEL_PATH=deepseek-ai/DeepSeek-R1-Distill-Qwen-7B
7
  HF_MODEL_NAME=$(basename ${HF_MODEL_PATH})
8
  LOCAL_MODEL_PATH=./data/${HF_MODEL_NAME}
9