diff --git a/Dockerfile b/Dockerfile
index 464b462..e2e0004 100644
--- a/Dockerfile
+++ b/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
diff --git a/requirements.txt b/requirements.txt
index 98c8d5e..9301bef 100644
--- 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