aloatist commited on
Commit
1aa8238
·
verified ·
1 Parent(s): a08bcbf

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +7 -3
Dockerfile CHANGED
@@ -43,11 +43,15 @@ 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 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/}"]
 
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
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='umt5_xxl_fp8_e4m3fn_scaled.safetensors', subfolder='split_files/text_encoders', local_dir='./models/text_encoders'); \
50
+ 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'); \
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
+ hf_hub_download(repo_id='Comfy-Org/Wan_2.1_ComfyUI_repackaged', filename='image_to_video_wan_720p_example.json', subfolder='example%20workflows_Wan2.1', local_dir='./user/default/workflows')"
53
 
54
+ # Tải workflow từ foxtonai
55
+ RUN mkdir -p ./user/default/workflows && \
56
+ wget -P ./user/default/workflows https://foxtonai.com/downloads/comfyui-wan-2.1-i2v-modified-workflow.json
57
  CMD ["python", "main.py", "--listen", "0.0.0.0", "--cpu", "--port", "7860", "--use-split-cross-attention", "--output-directory", "${USE_PERSISTENT_DATA:+/data/}"]