koai / Dockerfile
kines9661's picture
Update Dockerfile
f7930cd verified
raw
history blame contribute delete
503 Bytes
FROM python:3.10-slim
WORKDIR /app
# 更新 pip
RUN pip install --upgrade pip
# 分步安裝避免衝突
RUN pip install torch==2.0.1 torchvision==0.15.2 --index-url https://download.pytorch.org/whl/cpu
RUN pip install diffusers==0.25.1 transformers==4.35.0
RUN pip install gradio==4.28.3 fastapi==0.104.0 uvicorn==0.23.2
RUN pip install accelerate==0.24.0 safetensors==0.4.0 Pillow==10.0.1
RUN pip install pydantic==2.4.2 huggingface-hub==0.19.0 requests==2.31.0
COPY . .
CMD ["python", "app.py"]