hafidhsoekma commited on
Commit
240e4f4
1 Parent(s): 4994e9f

Update entrypoint.sh.template

Browse files
Files changed (1) hide show
  1. entrypoint.sh.template +1 -0
entrypoint.sh.template CHANGED
@@ -4,6 +4,7 @@
4
  mongod &
5
 
6
  # Start the text-generation-inference process
 
7
  text-generation-launcher --model-id ${MODEL_NAME} --quantize gptq --num-shard 1 --port 8080 --trust-remote-code &
8
 
9
  # Wait for text-generation-inference to start
 
4
  mongod &
5
 
6
  # Start the text-generation-inference process
7
+ pip3 install -U transformers && pip install -U transformers
8
  text-generation-launcher --model-id ${MODEL_NAME} --quantize gptq --num-shard 1 --port 8080 --trust-remote-code &
9
 
10
  # Wait for text-generation-inference to start