Spaces:
Running
Running
File size: 828 Bytes
d2d260a eae75bd d2d260a eae75bd 60ae242 eae75bd |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 |
FROM rocm/vllm:latest
# Set environment variables to reduce prompts and enable UTF-8
ENV DEBIAN_FRONTEND=noninteractive
ENV PYTHONDONTWRITEBYTECODE=1
ENV PYTHONUNBUFFERED=1
# Install system dependencies
RUN apt-get update && apt-get install -y wget && rm -rf /var/lib/apt/lists/*
# Upgrade pip and install Python packages
RUN pip install --upgrade pip && \
pip install diffusers peft==0.17.0 matplotlib jupyter transformers==4.56.1 gradio && \
pip install huggingface_hub
RUN git clone https://github.com/ROCm/flash-attention.git && cd flash-attention && GPU_ARCHS=gfx942 python setup.py install
RUN pip install spaces easydict ftfy imageio
RUN git clone https://huggingface.co/spaces/vivienfanghua/wan2.2_enhanced_amd && cd wan2.2_enhanced_amd
WORKDIR /app/wan2.2_enhanced_amd
ENTRYPOINT [ "python", "app.py"]
|