chansung commited on
Commit
97dc701
1 Parent(s): 0f79948

Update entrypoint.sh.template

Browse files
Files changed (1) hide show
  1. entrypoint.sh.template +2 -2
entrypoint.sh.template CHANGED
@@ -1,8 +1,8 @@
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
  fi
 
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
  fi