Minimal deps + only essential RVC modules

Dockerfile

@@ -1,4 +1,4 @@
-FROM pytorch/pytorch:2.4.1-cpu
+FROM python:3.10-slim
 
 ENV DEBIAN_FRONTEND=noninteractive
 ENV PIP_NO_CACHE_DIR=1
@@ -14,13 +14,20 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
 
 WORKDIR /app
 
-# Install remaining Python dependencies (torch/torchaudio already in base)
+# Install minimal Python dependencies in one go
 COPY requirements.txt .
 RUN pip install --no-cache-dir --timeout=300 -r requirements.txt
 
-# Copy RVC source code
-COPY infer ./infer
-COPY tools ./tools
+# Copy only essential RVC modules
+COPY infer/__init__.py infer/
+COPY infer/audio.py infer/
+COPY infer/hubert.py infer/
+COPY infer/rmvpe.py infer/
+COPY infer/fcpe.py infer/
+COPY infer/vc ./infer/vc
+COPY infer/module ./infer/module
+COPY tools/cuda_graph.py tools/
+COPY tools/file_io.py tools/
 COPY configs ./configs
 COPY i18n ./i18n
 

requirements.txt

@@ -1,4 +1,8 @@
-# Runtime dependencies for RVC inference (torch/torchaudio in base image)
+# Minimal RVC inference dependencies
+torch==2.4.1+cpu
+torchaudio==2.4.1+cpu
+--index-url https://download.pytorch.org/whl/cpu
+
 numpy>=1.24.0,<2
 scipy>=1.10.0,<2
 librosa>=0.10.0,<0.11
@@ -10,19 +14,10 @@ tqdm>=4.67.0,<5
 huggingface_hub>=0.20.0
 ffmpeg-python>=0.2.0,<1
 av>=15.1.0,<16
-einops>=0.8.0,<1
 onnxruntime>=1.24.4,<2
-opencv-python-headless>=4.10.0,<5
 scikit-learn>=1.6.0,<2
 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
 
-# Gradio & web deps
 gradio>=4.44.0
 fastapi>=0.88.0,<0.100
 uvicorn>=0.20.0,<0.23