Modify Dockerfile and requirements.txt.

Dockerfile

@@ -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

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