Spaces:
Sleeping
Sleeping
echo "MODEL_NAME: $MODEL_NAME" | |
echo "QUANTIZE: $QUANTIZE" | |
echo "GRADIO_PORT: $GRADIO_PORT" | |
# Start the text-generation-inference process | |
if [[ -n "$QUANTIZE" && "$QUANTIZE" != "false" ]]; then | |
text-generation-launcher --model-id $MODEL_NAME --num-shard 1 --port 8080 --trust-remote-code --quantize $QUANTIZE & | |
else | |
text-generation-launcher --model-id $MODEL_NAME --num-shard 1 --port 8080 --trust-remote-code & | |
fi | |
# Wait for text-generation-inference to start | |
curl --retry 60 --retry-delay 10 --retry-connrefused http://127.0.0.1:8080/health | |
# Start the gradio | |
python app/main.py --port $GRADIO_PORT & | |
# Wait for any process to exit | |
wait -n | |
# Exit with status of process that exited first | |
exit $? |