Selaa lähdekoodia

Pin transformers version (#90)

Refs #88
AbdBarho 4 vuotta sitten
vanhempi
sitoutus
556a50f49b
3 muutettua tiedostoa jossa 4 lisäystä ja 3 poistoa
  1. 2 1
      services/AUTOMATIC1111/Dockerfile
  2. 1 1
      services/hlky/Dockerfile
  3. 1 1
      services/lstein/Dockerfile

+ 2 - 1
services/AUTOMATIC1111/Dockerfile

@@ -47,12 +47,13 @@ RUN pip install --prefer-binary --no-cache-dir -r ${ROOT}/repositories/CodeForme
 # Note: don't update the sha of previous versions because the install will take forever
 # instead, update the repo state in a later step
 
-ARG SHA=d6fd71f36f33763f3a8d1d98f815e1e6a979e13e
+ARG SHA=79e7c392989ad70a1c02cbfe6eb38ee5a78bdbce
 RUN <<EOF
 cd stable-diffusion-webui
 git pull --rebase
 git reset --hard ${SHA}
 pip install --prefer-binary --no-cache-dir -r requirements.txt
+pip install --prefer-binary --no-cache-dir -r requirements_versions.txt
 EOF
 
 RUN pip install --prefer-binary -U --no-cache-dir opencv-python-headless

+ 1 - 1
services/hlky/Dockerfile

@@ -34,7 +34,7 @@ conda env update --file environment.yaml -n base
 conda clean -a -y
 EOF
 
-RUN pip uninstall transformers -y && pip install -U --no-cache-dir pyperclip transformers==4.22
+RUN pip install -U --no-cache-dir pyperclip
 
 # Latent diffusion
 RUN <<EOF

+ 1 - 1
services/lstein/Dockerfile

@@ -37,7 +37,7 @@ conda env update --file environment.yaml -n base
 conda clean -a -y
 EOF
 
-RUN pip uninstall opencv-python -y && pip install --prefer-binary --force-reinstall --no-cache-dir opencv-python-headless transformers==4.22
+RUN pip uninstall opencv-python -y && pip install --prefer-binary --force-reinstall --no-cache-dir opencv-python-headless transformers==4.19.2
 
 COPY . /docker/
 RUN <<EOF