Set transformers to version 4.42.4

ashleykleynhans · Jul 30, 2024 · f44b155 · f44b155
1 parent 7ce77b1
commit f44b155
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/Dockerfile b/Dockerfile
@@ -35,12 +35,12 @@ COPY a1111/cache-sd-model.py ./
 
 # Cache Base Model
 RUN source /venv/bin/activate && \
-    python3 cache-sd-model.py --skip-torch-cuda-test --use-cpu=all --xformers --ckpt /sd-models/sd_xl_base_1.0.safetensors && \
+    python3 cache-sd-model.py --skip-install --skip-torch-cuda-test --use-cpu=all --xformers --ckpt /sd-models/sd_xl_base_1.0.safetensors && \
     deactivate
 
 # Cache Refiner Model
 RUN source /venv/bin/activate && \
-    python3 cache-sd-model.py --skip-torch-cuda-test --use-cpu=all --xformers --ckpt /sd-models/sd_xl_refiner_1.0.safetensors && \
+    python3 cache-sd-model.py --skip-install --skip-torch-cuda-test --use-cpu=all --xformers --ckpt /sd-models/sd_xl_refiner_1.0.safetensors && \
     deactivate
 
 # RUN cd /stable-diffusion-webui && python cache.py --use-cpu=all --ckpt /model.safetensors

diff --git a/build/install_a1111.sh b/build/install_a1111.sh
@@ -70,7 +70,7 @@ cd /stable-diffusion-webui/extensions/sd-webui-controlnet
 pip3 install -r requirements.txt
 
 # Install newer version of transformers so that SD3 model can be loaded
-pip3 install transformers>=4.43.3
+pip3 install transformers==4.42.4
 pip3 cache purge
 
 deactivate