rabak1

Dockerfile

@@ -27,8 +27,9 @@ RUN pip install --no-cache-dir --upgrade pip setuptools wheel
 # Install NumPy with version that satisfies all dependencies
 RUN pip install --no-cache-dir "numpy>=1.23.5,<1.24"
 
-# CPU-only PyTorch (install early to avoid memory issues)
-RUN pip install --no-cache-dir torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu
+# CPU-only PyTorch (use version compatible with TTS library)
+# PyTorch 2.6+ breaks TTS due to weights_only=True default, so use 2.5.1
+RUN pip install --no-cache-dir torch==2.5.1 torchvision==0.20.1 torchaudio==2.5.1 --index-url https://download.pytorch.org/whl/cpu
 
 # Core dependencies - install in smaller batches
 RUN pip install --no-cache-dir scipy==1.11.3
@@ -79,4 +80,4 @@ RUN cd /app/SadTalker && \
 EXPOSE 8501
 CMD ["streamlit", "run", "avatar_streamlit.py", "--server.port=8501", "--server.address=0.0.0.0"]
 
-# FORCE_REBUILD: 2025-01-04-v4
+# FORCE_REBUILD: 2025-01-04-v5