Upload Dockerfile

Dockerfile

@@ -0,0 +1,56 @@
+# 1. 使用与 README 要求一致的 PyTorch 版本 (Torch 2.4.1)
+FROM pytorch/pytorch:2.4.1-cuda12.1-cudnn9-devel
+
+# 设置非交互式安装
+ENV DEBIAN_FRONTEND=noninteractive
+
+# 2. 安装系统依赖 (README 要求安装 FFmpeg)
+RUN apt-get update && apt-get install -y \
+    git \
+    git-lfs \
+    ffmpeg \
+    libgl1-mesa-glx \
+    libglib2.0-0 \
+    && rm -rf /var/lib/apt/lists/*
+
+# 设置工作目录
+WORKDIR /app
+
+# 3. 克隆 InfiniteTalk 仓库
+RUN git clone https://github.com/MeiGen-AI/InfiniteTalk.git
+
+# 进入仓库目录
+WORKDIR /app/InfiniteTalk
+
+# 4. 安装 Python 依赖 (严格按照 README 顺序)
+# 先升级 pip
+RUN pip install --no-cache-dir --upgrade pip
+
+# 安装 PyTorch 相关 (基础镜像已有，但为了确保 xformers 匹配，我们可以再次确认或跳过)
+# README: pip install -U xformers==0.0.28 --index-url https://download.pytorch.org/whl/cu121
+RUN pip install -U xformers==0.0.28 --index-url https://download.pytorch.org/whl/cu121
+
+# 安装 Flash Attention 及其依赖 (README 步骤 2)
+# misaki[en], ninja, psutil, packaging, wheel
+RUN pip install --no-cache-dir misaki[en] ninja psutil packaging wheel
+# 安装 flash-attn
+RUN pip install flash_attn==2.7.4.post1
+
+# 安装其他依赖 (README 步骤 3)
+RUN pip install --no-cache-dir -r requirements.txt
+# 补充一些 Space 运行需要的库
+RUN pip install --no-cache-dir gradio huggingface_hub
+
+# 5. 设置权限 (Hugging Face Space 安全要求)
+RUN useradd -m -u 1000 user
+RUN chown -R user:user /app
+USER user
+
+# 6. 设置 Gradio 环境变量，确保可以通过公网访问
+ENV GRADIO_SERVER_NAME="0.0.0.0"
+
+# 7. 复制我们的启动脚本 (下一步创建)
+COPY --chown=user:user run.py /app/InfiniteTalk/run.py
+
+# 8. 启动
+CMD ["python", "run.py"]