Create Dockerfile

Dockerfile

@@ -0,0 +1,25 @@
+# HF Spaces 官方基础镜像，匹配CUDA
+FROM hf-mirror.com/huggingface/transformers-pytorch-gpu:latest
+
+# 1. 预装编译flash-attn必需的系统依赖
+RUN apt-get update && apt-get install -y --no-install-recommends \
+    gcc g++ build-essential ninja-build cuda-nvcc-12-1 \
+    && rm -rf /var/lib/apt/lists/*
+
+# 升级基础打包工具
+RUN pip install --upgrade pip setuptools wheel ninja
+
+# 2. 复制依赖并安装，强制源码编译flash-attn，规避wheel不兼容
+COPY requirements.txt /tmp/requirements.txt
+RUN MAX_JOBS=2 pip install \
+    --no-cache-dir \
+    --no-binary flash-attn \
+    --no-build-isolation \
+    -r /tmp/requirements.txt \
+    gradio[oauth]==5.25.2 \
+    "uvicorn>=0.14.0" \
+    "websockets>=10.4" \
+    spaces
+
+# HF Spaces固定启动命令，不要修改
+CMD ["python", "app.py"]