Spaces:

roll-ai
/

RealCam-I2V

Runtime error

roll-ai commited on 4 days ago

Commit

ad9db23

verified ·

1 Parent(s): a1fa1f0

Update Dockerfile

Files changed (1) hide show

Dockerfile CHANGED Viewed

@@ -38,7 +38,7 @@ COPY requirements.txt .
 # Pre-install torch so flash-attn doesn't fail
 RUN pip install --upgrade pip && \
-    pip install torch==2.6.0 torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu121
 # Install the rest of the dependencies
 RUN pip install --no-cache-dir -r requirements.txt

 # Pre-install torch so flash-attn doesn't fail
 RUN pip install --upgrade pip && \
+    pip install torch==2.6 torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu121
 # Install the rest of the dependencies
 RUN pip install --no-cache-dir -r requirements.txt