remove vision from req add to docker

JEMeyer · Jan 5, 2025 · 5fa3e30 · 5fa3e30
1 parent e2fae01
commit 5fa3e30
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git a/Dockerfile b/Dockerfile
@@ -22,7 +22,7 @@ WORKDIR /app
 
 # Install PyTorch separately with CUDA 12.1
 # Ensure 'torch' is **not** listed in requirements.txt to avoid duplication
-RUN /opt/venv/bin/pip install torch==2.0.1+cu121 --extra-index-url https://download.pytorch.org/whl/cu121
+RUN /opt/venv/bin/pip install torch==2.0.1+cu121 torchvision --extra-index-url https://download.pytorch.org/whl/cu121
 
 # Copy and install Python dependencies
 COPY requirements.txt .

diff --git a/requirements.txt b/requirements.txt
@@ -3,5 +3,4 @@ uvicorn[standard]==0.22.0
 diffusers==0.20.2
 transformers==4.31.0
 Pillow==10.0.0
-xformers==0.0.20
-torchvision==0.15.2
+xformers==0.0.20\