Update Dockerfile

Dockerfile

@@ -39,14 +39,15 @@ RUN pyenv install $PYTHON_VERSION && \
 
 WORKDIR $HOME/app
 
-# Clone ComfyUI và cài đặt
+# Clone ComfyUI
 RUN git clone https://github.com/comfyanonymous/ComfyUI.git . && \
     pip install --no-cache-dir -r requirements.txt
 
-# Tải WAN 2.1 I2V-14B-720P
-RUN wget -c https://huggingface.co/Comfy-Org/Wan_2.1_ComfyUI_repackaged/resolve/main/wan2.1_i2v_720p_14B_fp8_e4m3fn.safetensors -P ./models/diffusion_models/ && \
-    wget -c https://huggingface.co/Comfy-Org/Wan_2.1_ComfyUI_repackaged/resolve/main/wan_2.1_vae.safetensors -P ./models/vae/ && \
-    wget -c https://huggingface.co/Comfy-Org/Wan_2.1_ComfyUI_repackaged/resolve/main/umt5_xxl_fp8_e4m3fn_scaled.safetensors -P ./models/text_encoders/ && \
-    wget -c https://huggingface.co/Comfy-Org/Wan_2.1_ComfyUI_repackaged/resolve/main/clip_vision_h.safetensors -P ./models/clip_vision/
+# Tải WAN 2.1 I2V-14B-720P bằng huggingface_hub
+RUN python -c "from huggingface_hub import hf_hub_download; \
+    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'); \
+    hf_hub_download(repo_id='Comfy-Org/Wan_2.1_ComfyUI_repackaged', filename='wan_2.1_vae.safetensors', local_dir='./models/vae'); \
+    hf_hub_download(repo_id='Comfy-Org/Wan_2.1_ComfyUI_repackaged', filename='umt5_xxl_fp8_e4m3fn_scaled.safetensors', local_dir='./models/text_encoders'); \
+    hf_hub_download(repo_id='Comfy-Org/Wan_2.1_ComfyUI_repackaged', filename='clip_vision_h.safetensors', local_dir='./models/clip_vision')"
 
 CMD ["python", "main.py", "--listen", "0.0.0.0", "--cpu", "--port", "7860", "--use-split-cross-attention", "--output-directory", "${USE_PERSISTENT_DATA:+/data/}"]