vivienfanghua commited on
Commit
eae75bd
·
1 Parent(s): d2d260a
Files changed (1) hide show
  1. Dockerfile +8 -2
Dockerfile CHANGED
@@ -13,6 +13,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
 
 
 
 
 
 
 
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 && GPU_ARCHS=gfx942 python setup.py install
17
 
18
+ RUN pip install spaces easydict ftfy imageio
19
+
20
+ RUN git clone https://huggingface.co/spaces/vivienfanghua/wan2.2_enhanced_amd && cd wan2.2_enhanced_amd &&
21
+
22
+ WORKDIR /app/wan2.2_enhanced_amd
23
+
24
+ ENTRYPOINT [ "python", "app.py"]