From f44b15533dfda63c79e020849df924242e3f8a15 Mon Sep 17 00:00:00 2001 From: Ashley Kleynhans Date: Wed, 31 Jul 2024 01:51:22 +0200 Subject: [PATCH] Set transformers to version 4.42.4 --- Dockerfile | 4 ++-- build/install_a1111.sh | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/Dockerfile b/Dockerfile index 4bb7564..75bbf31 100644 --- 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 index ab07e85..405ed8f 100644 --- 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