Update entrypoint.sh.template

entrypoint.sh.template

@@ -1,21 +1,15 @@
 #!/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 &
+if [[ "$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 &
 
 # 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 &
-ls -ltr
+# Start the gradio
 python app/main.py &
 
 # Wait for any process to exit