Spaces:
Running
Running
Upload folder using huggingface_hub
Browse files- Dockerfile +18 -0
- app.py +1 -1
Dockerfile
ADDED
|
@@ -0,0 +1,18 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
FROM rocm/vllm:latest
|
| 2 |
+
|
| 3 |
+
# Set environment variables to reduce prompts and enable UTF-8
|
| 4 |
+
ENV DEBIAN_FRONTEND=noninteractive
|
| 5 |
+
ENV PYTHONDONTWRITEBYTECODE=1
|
| 6 |
+
ENV PYTHONUNBUFFERED=1
|
| 7 |
+
|
| 8 |
+
# Install system dependencies
|
| 9 |
+
RUN apt-get update && apt-get install -y wget && rm -rf /var/lib/apt/lists/*
|
| 10 |
+
|
| 11 |
+
# Upgrade pip and install Python packages
|
| 12 |
+
RUN pip install --upgrade pip && \
|
| 13 |
+
pip install diffusers peft==0.17.0 matplotlib jupyter transformers==4.56.1 gradio && \
|
| 14 |
+
pip install huggingface_hub
|
| 15 |
+
|
| 16 |
+
RUN git clone https://github.com/ROCm/flash-attention.git && cd flash-attention && python setup.py
|
| 17 |
+
|
| 18 |
+
RUN git clone https://huggingface.co/spaces/vivienfanghua/wan2.2_enhanced_amd && cd wan2.2_enhanced_amd && python3 app.py
|
app.py
CHANGED
|
@@ -384,4 +384,4 @@ with gr.Blocks(css=css, theme=gr.themes.Soft(), delete_cache=(60, 900)) as demo:
|
|
| 384 |
)
|
| 385 |
|
| 386 |
if __name__ == "__main__":
|
| 387 |
-
demo.launch()
|
|
|
|
| 384 |
)
|
| 385 |
|
| 386 |
if __name__ == "__main__":
|
| 387 |
+
demo.launch(server_name="0.0.0.0", server_port=7860)
|