aloatist commited on
Commit
257e595
·
verified ·
1 Parent(s): fd9f1e7

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +5 -5
Dockerfile CHANGED
@@ -43,11 +43,11 @@ WORKDIR $HOME/app
43
  RUN git clone https://github.com/comfyanonymous/ComfyUI.git . && \
44
  pip install --no-cache-dir -r requirements.txt
45
 
46
- # Tải WAN 2.1 I2V-14B-720P bằng huggingface_hub
47
  RUN python -c "from huggingface_hub import hf_hub_download; \
48
- hf_hub_download(repo_id='Comfy-Org/Wan_2.1_ComfyUI_repackaged', filename='wan2.1_i2v_720p_14B_fp8_e4m3fn.safetensors', local_dir='./models/diffusion_models'); \
49
- hf_hub_download(repo_id='Comfy-Org/Wan_2.1_ComfyUI_repackaged', filename='wan_2.1_vae.safetensors', local_dir='./models/vae'); \
50
- hf_hub_download(repo_id='Comfy-Org/Wan_2.1_ComfyUI_repackaged', filename='umt5_xxl_fp8_e4m3fn_scaled.safetensors', local_dir='./models/text_encoders'); \
51
- hf_hub_download(repo_id='Comfy-Org/Wan_2.1_ComfyUI_repackaged', filename='clip_vision_h.safetensors', local_dir='./models/clip_vision')"
52
 
53
  CMD ["python", "main.py", "--listen", "0.0.0.0", "--cpu", "--port", "7860", "--use-split-cross-attention", "--output-directory", "${USE_PERSISTENT_DATA:+/data/}"]
 
43
  RUN git clone https://github.com/comfyanonymous/ComfyUI.git . && \
44
  pip install --no-cache-dir -r requirements.txt
45
 
46
+ # Tải WAN 2.1 I2V-14B-720P bằng huggingface_hub với subfolder
47
  RUN python -c "from huggingface_hub import hf_hub_download; \
48
+ hf_hub_download(repo_id='Comfy-Org/Wan_2.1_ComfyUI_repackaged', filename='wan2.1_i2v_720p_14B_fp8_e4m3fn.safetensors', subfolder='split_files/diffusion_models', local_dir='./models/diffusion_models'); \
49
+ hf_hub_download(repo_id='Comfy-Org/Wan_2.1_ComfyUI_repackaged', filename='wan_2.1_vae.safetensors', subfolder='split_files/vae', local_dir='./models/vae'); \
50
+ hf_hub_download(repo_id='Comfy-Org/Wan_2.1_ComfyUI_repackaged', filename='umt5_xxl_fp8_e4m3fn_scaled.safetensors', subfolder='split_files/text_encoders', local_dir='./models/text_encoders'); \
51
+ hf_hub_download(repo_id='Comfy-Org/Wan_2.1_ComfyUI_repackaged', filename='clip_vision_h.safetensors', subfolder='split_files/clip_vision', local_dir='./models/clip_vision')"
52
 
53
  CMD ["python", "main.py", "--listen", "0.0.0.0", "--cpu", "--port", "7860", "--use-split-cross-attention", "--output-directory", "${USE_PERSISTENT_DATA:+/data/}"]