XanderJC commited on
Commit
23f9321
·
1 Parent(s): 05e15ac
Files changed (1) hide show
  1. Dockerfile +6 -1
Dockerfile CHANGED
@@ -27,4 +27,9 @@ RUN pip install vllm==0.7.2
27
  # breaking changes in master
28
  RUN pip install --upgrade git+https://github.com/huggingface/transformers.git@336dc69d63d56f232a183a3e7f52790429b871ef
29
 
30
- ENTRYPOINT ["sh", "-c", "python -m vllm.entrypoints.openai.api_server --model convergence-ai/proxy-lite --trust-remote-code --tokenizer-pool-size 10 --limit-mm-per-prompt image=2 --enable-auto-tool-choice --tool-call-parser hermes --port 7860"]
 
 
 
 
 
 
27
  # breaking changes in master
28
  RUN pip install --upgrade git+https://github.com/huggingface/transformers.git@336dc69d63d56f232a183a3e7f52790429b871ef
29
 
30
+ RUN pip install ray[adag]
31
+ RUN pip install cupy
32
+
33
+ RUN export VLLM_USE_V1=1
34
+
35
+ ENTRYPOINT ["sh", "-c", "python -m vllm.entrypoints.openai.api_server --model convergence-ai/proxy-lite --trust-remote-code --tokenizer-pool-size 20 --limit-mm-per-prompt image=1 --enable-auto-tool-choice --tool-call-parser hermes --port 7860"]