| FROM python:3.11-slim | |
| # v6 force rebuild - 简化依赖安装 | |
| WORKDIR /app | |
| # 安装依赖 (v6 - 分步安装避免超时) | |
| RUN pip install --no-cache-dir fastapi==0.104.1 uvicorn==0.24.0 numpy==1.24.4 | |
| RUN pip install --no-cache-dir torch --index-url https://download.pytorch.org/whl/cpu | |
| RUN pip install --no-cache-dir huggingface_hub==0.20.1 | |
| # 复制源代码 | |
| COPY src/ ./src/ | |
| COPY data/ ./data/ | |
| COPY models/ ./models/ | |
| COPY train_200m.py . | |
| COPY app.py . | |
| # 创建目录 | |
| RUN mkdir -p /app/checkpoints /app/models /app/tools | |
| ENV PYTHONPATH=/app | |
| ENV PYTHONUNBUFFERED=1 | |
| ENV SWARM_PRODUCT_ROOT=/app | |
| ENV HF_MODEL_REPO=lk080424/swarm-weights | |
| ENV ENABLE_TRAINING=1 | |
| EXPOSE 7860 | |
| CMD ["python", "app.py"] | |