cyfyifanchen commited on
Commit
7f0fc1a
·
verified ·
1 Parent(s): a2a7600

Update entrypoint.sh

Browse files
Files changed (1) hide show
  1. entrypoint.sh +6 -6
entrypoint.sh CHANGED
@@ -12,13 +12,13 @@ mkdir -p data
12
  # vllm serve ${LOCAL_MODEL_PATH_1} --served-model-name ${HF_MODEL_NAME_1} --api-key ${VLLM_SERVE_API_KEY_1} &
13
 
14
  # Shisa 7b V1
15
- HF_MODEL_PATH_2=TEN-framework/Shisa-7b-V1
16
- HF_MODEL_NAME_2=$(basename ${HF_MODEL_PATH_2})
17
- LOCAL_MODEL_PATH_2=./data/${HF_MODEL_NAME_2}
18
- VLLM_SERVE_API_KEY_2=Shisa_7b_V1
19
 
20
- HF_HUB_ENABLE_HF_TRANSFER=1 huggingface-cli download ${HF_MODEL_PATH_2} --local-dir ${LOCAL_MODEL_PATH_2}
21
- vllm serve ${LOCAL_MODEL_PATH_2} --served-model-name ${HF_MODEL_NAME_2} --api-key ${VLLM_SERVE_API_KEY_2} &
22
 
23
  # Start frontend
24
  export NEXT_PUBLIC_EDIT_GRAPH_MODE=false
 
12
  # vllm serve ${LOCAL_MODEL_PATH_1} --served-model-name ${HF_MODEL_NAME_1} --api-key ${VLLM_SERVE_API_KEY_1} &
13
 
14
  # Shisa 7b V1
15
+ HF_MODEL_PATH=augmxnt/shisa-7b-v1
16
+ HF_MODEL_NAME=$(basename ${HF_MODEL_PATH})
17
+ LOCAL_MODEL_PATH=./data/${HF_MODEL_NAME}
18
+ VLLM_SERVE_API_KEY=Shisa_7b_V1
19
 
20
+ HF_HUB_ENABLE_HF_TRANSFER=1 huggingface-cli download ${HF_MODEL_PATH} --local-dir ${LOCAL_MODEL_PATH}
21
+ vllm serve ${LOCAL_MODEL_PATH} --served-model-name ${HF_MODEL_NAME} --api-key ${VLLM_SERVE_API_KEY} --port 8000
22
 
23
  # Start frontend
24
  export NEXT_PUBLIC_EDIT_GRAPH_MODE=false