ten-agent-demo / entrypoint.sh
cyfyifanchen's picture
Adding Shisa
12bbaf2 verified
raw
history blame
968 Bytes
#!/bin/bash
mkdir -p data
# TEN Turn Detection
HF_MODEL_PATH_1=TEN-framework/TEN_Turn_Detection
HF_MODEL_NAME_1=$(basename ${HF_MODEL_PATH_1})
LOCAL_MODEL_PATH_1=./data/${HF_MODEL_NAME_1}
VLLM_SERVE_API_KEY_1=TEN_Turn_Detection
HF_HUB_ENABLE_HF_TRANSFER=1 huggingface-cli download ${HF_MODEL_PATH_1} --local-dir ${LOCAL_MODEL_PATH_1}
vllm serve ${LOCAL_MODEL_PATH_1} --served-model-name ${HF_MODEL_NAME_1} --api-key ${VLLM_SERVE_API_KEY_1} &
# Shisa 7b V1
HF_MODEL_PATH_2=TEN-framework/Shisa-7b-V1
HF_MODEL_NAME_2=$(basename ${HF_MODEL_PATH_2})
LOCAL_MODEL_PATH_2=./data/${HF_MODEL_NAME_2}
VLLM_SERVE_API_KEY_2=Shisa_7b_V1
HF_HUB_ENABLE_HF_TRANSFER=1 huggingface-cli download ${HF_MODEL_PATH_2} --local-dir ${LOCAL_MODEL_PATH_2}
vllm serve ${LOCAL_MODEL_PATH_2} --served-model-name ${HF_MODEL_NAME_2} --api-key ${VLLM_SERVE_API_KEY_2} &
# Start frontend
export NEXT_PUBLIC_EDIT_GRAPH_MODE=false
cd /app/demo && npm run dev &
# Start backend
cd /app && task run