Teddy-Project commited on
Commit
d76497a
·
verified ·
1 Parent(s): 467a3bb

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +1 -1
Dockerfile CHANGED
@@ -1,7 +1,7 @@
1
  FROM ghcr.io/huggingface/text-generation-inference:1.3
2
 
3
  ENV MODEL_ID=HuggingFaceH4/zephyr-7b-beta
4
- ENV MAX_BATCH_PREFILL_TOKENS=32000
5
  ENV MAX_INPUT_TOKENS=4096
6
  ENV MAX_TOTAL_TOKENS=8192
7
 
 
1
  FROM ghcr.io/huggingface/text-generation-inference:1.3
2
 
3
  ENV MODEL_ID=HuggingFaceH4/zephyr-7b-beta
4
+ ENV MAX_BATCH_PREFILL_TOKENS=24576
5
  ENV MAX_INPUT_TOKENS=4096
6
  ENV MAX_TOTAL_TOKENS=8192
7