Skip to content

Commit 9a6aac6

Browse files
committed
Update transformers and tokenizers in Dockerfile for CVE fix
Signed-off-by: heyselbi <[email protected]>
1 parent 9640dd8 commit 9a6aac6

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

Dockerfile

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -165,7 +165,7 @@ RUN cd server && \
165165
pip install ".[accelerate]" --no-cache-dir
166166

167167
# temp: install newer transformers lib that optimum clashes with
168-
RUN pip install transformers==4.40.0 tokenizers==0.19.1 --no-cache-dir
168+
RUN pip install transformers==4.48.0 tokenizers==0.21.1 --no-cache-dir
169169

170170
# Patch codegen model changes into transformers
171171
RUN cp server/transformers_patch/modeling_codegen.py ${SITE_PACKAGES}/transformers/models/codegen/modeling_codegen.py
@@ -292,7 +292,7 @@ COPY server server
292292
RUN cd server && make gen-server && pip install ".[accelerate, ibm-fms, onnx-gpu, quantize]" --no-cache-dir --extra-index-url=https://aiinfra.pkgs.visualstudio.com/PublicPackages/_packaging/onnxruntime-cuda-12/pypi/simple/
293293

294294
# temp: install newer transformers lib that optimum clashes with
295-
RUN pip install transformers==4.40.0 tokenizers==0.19.1 --no-cache-dir
295+
RUN pip install transformers==4.48.0 tokenizers==0.21.1 --no-cache-dir
296296

297297
# Patch codegen model changes into transformers 4.35
298298
RUN cp server/transformers_patch/modeling_codegen.py ${SITE_PACKAGES}/transformers/models/codegen/modeling_codegen.py

0 commit comments

Comments
 (0)