wangyoucao577 commited on
Commit
aa7ea36
·
1 Parent(s): 62d5680

feat: serve

Browse files
Files changed (1) hide show
  1. entrypoint.sh +9 -1
entrypoint.sh CHANGED
@@ -2,4 +2,12 @@
2
 
3
  mkdir -p data
4
 
5
- HF_HUB_ENABLE_HF_TRANSFER=1 huggingface-cli download Qwen/Qwen2.5-1.5B-Instruct --local-dir ./data/Qwen2.5-1.5B-Instruct
 
 
 
 
 
 
 
 
 
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
+
9
+ # download model
10
+ HF_HUB_ENABLE_HF_TRANSFER=1 huggingface-cli download ${HF_MODEL_PATH} --local-dir ${LOCAL_MODEL_PATH}
11
+
12
+ # start vllm server
13
+ ${HOME}/.local/bin/vllm serve ${LOCAL_MODEL_PATH} --served-model-name ${HF_MODEL_NAME}