Add optimized Dockerfile with pre-built mmcv-full wheel

.dockerignore

@@ -0,0 +1,29 @@
+.git
+.gitignore
+README.md
+Dockerfile
+.dockerignore
+__pycache__
+*.pyc
+*.pyo
+*.pyd
+.Python
+env
+pip-log.txt
+pip-delete-this-directory.txt
+.tox
+.coverage
+.coverage.*
+.cache
+nosetests.xml
+coverage.xml
+*.cover
+*.log
+.git
+.mypy_cache
+.pytest_cache
+.hypothesis
+
+.DS_Store
+.vscode
+.idea

Dockerfile

@@ -0,0 +1,41 @@
+FROM python:3.10-slim
+
+# Install system dependencies
+RUN apt-get update && apt-get install -y \
+    ffmpeg \
+    git \
+    wget \
+    curl \
+    build-essential \
+    && rm -rf /var/lib/apt/lists/*
+
+# Set working directory
+WORKDIR /home/user/app
+
+# Copy requirements first for better caching
+COPY requirements.txt .
+
+# Install PyTorch first
+RUN pip install --no-cache-dir torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu
+
+# Install mmcv-full with pre-built wheel (this is the key optimization)
+RUN pip install --no-cache-dir mmcv-full -f https://download.openmmlab.com/mmcv/dist/cpu/torch2.0/index.html
+
+# Install other dependencies
+RUN pip install --no-cache-dir -r requirements.txt
+
+# Copy the application code
+COPY . .
+
+# Create necessary directories
+RUN mkdir -p models/musetalk models/musetalkV15 models/syncnet models/dwpose models/face-parse-bisent models/sd-vae models/whisper
+
+# Set environment variables
+ENV PYTHONPATH=/home/user/app
+ENV HF_HOME=/home/user/app/.cache/huggingface
+
+# Expose port
+EXPOSE 7860
+
+# Run the application
+CMD ["python", "app.py"]

requirements.txt

@@ -1,6 +1,3 @@
-torch>=2.0.0
-torchvision
-torchaudio
 diffusers==0.30.2
 accelerate==0.28.0
 numpy==1.23.5
@@ -19,7 +16,6 @@ ffmpeg-python
 moviepy
 tqdm
 Pillow
-mmcv-full
 mmpose
 mmdet
 mmengine

setup_environment.sh

@@ -0,0 +1,21 @@
+#!/bin/bash
+
+# Install system dependencies
+apt-get update && apt-get install -y \
+    ffmpeg \
+    git \
+    wget \
+    curl \
+    build-essential \
+    && rm -rf /var/lib/apt/lists/*
+
+# Install PyTorch first
+pip install --no-cache-dir torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu
+
+# Install mmcv-full with pre-built wheel
+pip install --no-cache-dir mmcv-full -f https://download.openmmlab.com/mmcv/dist/cpu/torch2.0/index.html
+
+# Install other dependencies
+pip install --no-cache-dir -r requirements.txt
+
+echo "Environment setup completed!"