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

feat: smaller model

Browse files
Files changed (1) hide show
  1. entrypoint.sh +1 -1
entrypoint.sh CHANGED
@@ -3,7 +3,7 @@
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
 
 
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-1.5B
7
  HF_MODEL_NAME=$(basename ${HF_MODEL_PATH})
8
  LOCAL_MODEL_PATH=./data/${HF_MODEL_NAME}
9