vivienfanghua commited on
Commit
d2d260a
·
verified ·
1 Parent(s): 3c27b35

Upload folder using huggingface_hub

Browse files
Files changed (2) hide show
  1. Dockerfile +18 -0
  2. 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)