Spaces:
Running
on
A100
Running
on
A100
File size: 1,048 Bytes
783867d f2a7e09 fe7741a 783867d |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 |
#!/bin/bash
# Make sure `/data/db` directory exists even with persistent storage
mkdir -p /data/db
# If app crashed, mongo didn't stop gracefully. Remove all the old *.lock files
find /data/db -name "*.lock" -type f -exec rm -f {} \;
# Start the local Mongo database
mongod &
# Start the text-generation-inference process
text-generation-launcher --model-id ${MODEL_NAME} --num-shard 1 --port 8080 --trust-remote-code --shard-uds-path /tmp/text-generation-model-a --master-port 25000 --cuda-memory-fraction 0.5 &
text-generation-launcher --model-id HuggingFaceH4/mistral-7b-constitutional-ai --num-shard 1 --port 8081 --trust-remote-code --shard-uds-path /tmp/text-generation-model-b --master-port 25001 --cuda-memory-fraction 0.5 &
# Wait for text-generation-inference to start
curl --retry 60 --retry-delay 10 --retry-connrefused http://127.0.0.1:8080/health
# Start the chat-ui process
pm2 start /app/build/index.js -i $CPU_CORES --no-daemon &
# Wait for any process to exit
wait -n
# Exit with status of process that exited first
exit $? |