Spaces:
Restarting
on
L4
Restarting
on
L4
fix: api key
Browse files- 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
|