Spaces:
Sleeping
Sleeping
File size: 1,178 Bytes
433b752 2faeeb9 6cb335f a47da2f 2eb9393 a47da2f 6cb335f 7877569 cd75e75 6cb335f cd75e75 6cb335f 2faeeb9 433b752 a47da2f cfffe9e a47da2f bac467e 433b752 e11772a 433b752 6dee22b f28f0b8 433b752 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 |
#!/bin/bash
echo "MODEL_NAME: $MODEL_NAME"
echo "QUANTIZE: $QUANTIZE"
echo "GRADIO_PORT: $GRADIO_PORT"
MODEL_NAME=$(echo $MODEL_NAME | sed 's/^[[:space:]]*//g;s/[[:space:]]*$//g')
QUANTIZE=$(echo $QUANTIZE | sed 's/^[[:space:]]*//g;s/[[:space:]]*$//g')
GRADIO_PORT=$(echo $GRADIO_PORT | sed 's/^[[:space:]]*//g;s/[[:space:]]*$//g')
if [[ "$QUANTIZE" == "$QUANTIZE" ]]; then
echo "WTF1-1!"
else
echo "WTF1-2!!"
fi
if [[ "$QUANTIZE" == "false" ]]; then
echo "WTF2-1!"
else
echo "WTF2-2!!"
fi
if [[ "$QUANTIZE" == false ]]; then
echo "WTF3-1!"
else
echo "WTF3-2!!"
fi
# Start the text-generation-inference process
if [[ "$QUANTIZE" == "flase" ]]; then
text-generation-launcher --model-id $MODEL_NAME --num-shard 1 --port 8080 --trust-remote-code &
else
text-generation-launcher --model-id $MODEL_NAME --num-shard 1 --port 8080 --trust-remote-code --quantize $QUANTIZE &
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 $? |