wangyoucao577 commited on
Commit
db4e8bf
·
verified ·
1 Parent(s): 6fba259

fix: api key

Browse files
Files changed (1) hide show
  1. entrypoint.sh +5 -6
entrypoint.sh CHANGED
@@ -1,25 +1,24 @@
1
  #!/bin/bash
2
 
3
- # TODO: remove
4
- nvm --version
5
- node --version
6
- npm --version
7
-
8
  mkdir -p data
9
 
10
  HF_MODEL_PATH=Qwen/Qwen2.5-1.5B-Instruct
11
  HF_MODEL_NAME=$(basename ${HF_MODEL_PATH})
12
  LOCAL_MODEL_PATH=./data/${HF_MODEL_NAME}
13
 
 
 
 
14
  # download model
15
  HF_HUB_ENABLE_HF_TRANSFER=1 huggingface-cli download ${HF_MODEL_PATH} --local-dir ${LOCAL_MODEL_PATH}
16
 
17
  # start vllm server
18
- vllm serve ${LOCAL_MODEL_PATH} --served-model-name ${HF_MODEL_NAME} &
19
 
20
  # start frontend
21
  cd /app/playground && npm run dev &
22
 
23
  # start backend
24
  export OPENAI_MODEL=${HF_MODEL_NAME}
 
25
  cd /app && task run
 
1
  #!/bin/bash
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
+ # TODO: use your own key and put into secret
10
+ VLLM_SERVE_API_KEY=demo-123
11
+
12
  # download model
13
  HF_HUB_ENABLE_HF_TRANSFER=1 huggingface-cli download ${HF_MODEL_PATH} --local-dir ${LOCAL_MODEL_PATH}
14
 
15
  # start vllm server
16
+ vllm serve ${LOCAL_MODEL_PATH} --served-model-name ${HF_MODEL_NAME} --api-key ${VLLM_SERVE_API_KEY} &
17
 
18
  # start frontend
19
  cd /app/playground && npm run dev &
20
 
21
  # start backend
22
  export OPENAI_MODEL=${HF_MODEL_NAME}
23
+ export OPENAI_API_KEY=${VLLM_SERVE_API_KEY}
24
  cd /app && task run