Tomtom84 commited on
Commit
0543208
·
verified ·
1 Parent(s): 35b038c

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +2 -1
Dockerfile CHANGED
@@ -23,7 +23,8 @@ ENV VLLM_USE_LM_FORMAT_ENFORCER=0
23
  # torch==2.3.1+cu121 torchaudio==2.3.1+cu121 \
24
  # --index-url https://download.pytorch.org/whl/cu121
25
 
26
- RUN pip install --no-cache-dir vllm==0.4.2
 
27
 
28
  # Python-Abhängigkeiten
29
  COPY --chown=user requirements.txt .
 
23
  # torch==2.3.1+cu121 torchaudio==2.3.1+cu121 \
24
  # --index-url https://download.pytorch.org/whl/cu121
25
 
26
+ RUN pip install --no-cache-dir "transformers==4.40.2" "lm-format-enforcer==0.9.8"
27
+ RUN pip install --no-cache-dir vllm>=0.9.0
28
 
29
  # Python-Abhängigkeiten
30
  COPY --chown=user requirements.txt .