Modify Dockerfile and requirements.txt.

2023-06-08 21:08:18 +09:00 · 2023-06-08 21:08:18 +09:00 · 767f9f75fe
commit 767f9f75fe
parent 1b4c76902c
2 changed files with 6 additions and 8 deletions
--- a/2
+++ b/2
@ -2,4 +2,4 @@ FROM python:3.11.3-slim-bullseye
 COPY requirements.txt /
 RUN apt update \
    && apt install -y wget git \
-    && pip install -r requirements.txt --extra-index-url https://download.pytorch.org/whl/cu117 --pre xformers
+    && pip install -r requirements.txt --extra-index-url https://download.pytorch.org/whl/cu117
--- a/requirements.txt
+++ b/requirements.txt
@ -1,9 +1,7 @@
 accelerate
-scipy
-diffusers[torch]
-safetensors
+diffusers[torch]==0.16.1
+onnxruntime==1.15.0
+safetensors==0.3.1
 torch==2.0.1+cu117
-torchvision
-torchmetrics
-omegaconf
-transformers
+transformers==4.29.2
+xformers==0.0.20