Update entrypoint.sh.template
Browse files- entrypoint.sh.template +5 -11
entrypoint.sh.template
CHANGED
@@ -1,21 +1,15 @@
|
|
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 |
-
|
|
|
|
|
|
|
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
|
17 |
-
# pm2 start /app/build/index.js -i $CPU_CORES --no-daemon &
|
18 |
-
ls -ltr
|
19 |
python app/main.py &
|
20 |
|
21 |
# Wait for any process to exit
|
|
|
1 |
#!/bin/bash
|
2 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
3 |
# Start the text-generation-inference process
|
4 |
+
if [[ "$QUANTIZE" != "False" ]]; then
|
5 |
+
text-generation-launcher --model-id ${MODEL_NAME} --num-shard 1 --port 8080 --trust-remote-code --quantize ${$QUANTIZE} &
|
6 |
+
else
|
7 |
+
text-generation-launcher --model-id ${MODEL_NAME} --num-shard 1 --port 8080 --trust-remote-code &
|
8 |
|
9 |
# Wait for text-generation-inference to start
|
10 |
curl --retry 60 --retry-delay 10 --retry-connrefused http://127.0.0.1:8080/health
|
11 |
|
12 |
+
# Start the gradio
|
|
|
|
|
13 |
python app/main.py &
|
14 |
|
15 |
# Wait for any process to exit
|