Split pip install into batches to avoid memory issues

Dockerfile

@@ -4,24 +4,67 @@ ENV DEBIAN_FRONTEND=noninteractive
 ENV PIP_NO_CACHE_DIR=1
 ENV PYTHONUNBUFFERED=1
 
-# Install system dependencies with retry
-RUN for i in 1 2 3; do \
-      apt-get update && apt-get install -y --no-install-recommends \
-        ffmpeg \
-        libsndfile1 \
-        libportaudio2 \
-        libasound2 \
-        && rm -rf /var/lib/apt/lists/* \
-        && break || sleep 10; \
-    done
+# Install system dependencies
+RUN apt-get update && apt-get install -y --no-install-recommends \
+    ffmpeg \
+    libsndfile1 \
+    libportaudio2 \
+    libasound2 \
+    && rm -rf /var/lib/apt/lists/*
 
 WORKDIR /app
 
-# Install Python dependencies
+# Install Python dependencies in batches to avoid memory issues
 COPY requirements.txt .
-RUN pip install --no-cache-dir --timeout=300 -r requirements.txt
 
-# Copy RVC source code (pre-bundled to avoid git clone at build time)
+# Batch 1: Core ML deps (torch already has index-url)
+RUN pip install --no-cache-dir --timeout=300 \
+    torch==2.4.1+cpu torchaudio==2.4.1+cpu \
+    --index-url https://download.pytorch.org/whl/cpu
+
+# Batch 2: Audio processing
+RUN pip install --no-cache-dir --timeout=300 \
+    numpy>=1.24.0,<2 \
+    scipy>=1.10.0,<2 \
+    librosa>=0.10.0,<0.11 \
+    soundfile>=0.12.0,<1 \
+    ffmpeg-python>=0.2.0,<1 \
+    av>=15.1.0,<16
+
+# Batch 3: ML utilities
+RUN pip install --no-cache-dir --timeout=300 \
+    faiss-cpu>=1.13.0,<2 \
+    praat-parselmouth>=0.4.5,<1 \
+    transformers>=4.49.0,<4.50 \
+    onnxruntime>=1.24.4,<2 \
+    opencv-python-headless>=4.10.0,<5 \
+    scikit-learn>=1.6.0,<2 \
+    einops>=0.8.0,<1
+
+# Batch 4: Web UI
+RUN pip install --no-cache-dir --timeout=300 \
+    gradio>=4.44.0 \
+    fastapi>=0.88.0,<0.100 \
+    uvicorn>=0.20.0,<0.23 \
+    pydantic>=1.10.13,<2 \
+    httpx>=0.23.0,<0.24 \
+    anyio>=3.6.2,<4 \
+    starlette>=0.22.0,<0.28 \
+    websockets>=10.4,<11
+
+# Batch 5: Remaining
+RUN pip install --no-cache-dir --timeout=300 \
+    tqdm>=4.67.0,<5 \
+    huggingface_hub>=0.20.0 \
+    PyYAML>=6.0 \
+    coloredlogs>=15.0,<16 \
+    torchfcpe>=0.0.4,<0.1 \
+    hyper-connections>=0.1.8,<0.4.3 \
+    local-attention>=1.11.0,<2 \
+    networkx>=3.2.0,<4 \
+    matplotlib>=3.8.2,<4
+
+# Copy RVC source code
 COPY infer ./infer
 COPY tools ./tools
 COPY configs ./configs