Spaces:
Running
on
A100
Running
on
A100
Create entrypoint.sh
Browse files- entrypoint.sh +23 -0
entrypoint.sh
ADDED
@@ -0,0 +1,23 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
#!/bin/bash
|
2 |
+
|
3 |
+
# Make sure `/data/db` directory exists even with persistent storage
|
4 |
+
mkdir -p /data/db
|
5 |
+
# If app crashed, mongo didn't stop gracefully. Remove all the old *.lock files
|
6 |
+
find /data/db -name "*.lock" -type f -exec rm -f {} \;
|
7 |
+
# Start the local Mongo database
|
8 |
+
mongod &
|
9 |
+
|
10 |
+
# Start the text-generation-inference process
|
11 |
+
text-generation-launcher --model-id ${MODEL_NAME} --num-shard 1 --port 8080 --trust-remote-code &
|
12 |
+
|
13 |
+
# Wait for text-generation-inference to start
|
14 |
+
curl --retry 60 --retry-delay 10 --retry-connrefused http://127.0.0.1:8080/health
|
15 |
+
|
16 |
+
# Start the chat-ui process
|
17 |
+
pm2 start /app/build/index.js -i $CPU_CORES --no-daemon &
|
18 |
+
|
19 |
+
# Wait for any process to exit
|
20 |
+
wait -n
|
21 |
+
|
22 |
+
# Exit with status of process that exited first
|
23 |
+
exit $?
|