diff --git a/docker/build_image.py b/docker/build_image.py index bcbcee68..9d0fb0fa 100644 --- a/docker/build_image.py +++ b/docker/build_image.py @@ -208,6 +208,7 @@ class GPUImageBuilder(Builder): extra_content = """ RUN pip install tf-keras==2.16.0 --no-dependencies && \ pip install onnx==1.18.0 --no-dependencies && \ + pip install transformers==4.55.4 --no-dependencies && \ pip install --no-cache-dir torchsde jupyterlab torchmetrics==0.11.4 basicsr pynvml shortuuid && \ CUDA_HOME=/usr/local/cuda TORCH_CUDA_ARCH_LIST="6.0 6.1 7.0 7.5 8.0 8.6 8.9 9.0" \ pip install --no-cache-dir 'git+https://github.com/facebookresearch/detectron2.git'