app.py

# Tải WAN 2.1
def download_model(repo_id, filename, subfolder, target_subfolder):
    target_dir = os.path.join(MODEL_DIR, target_subfolder)
    os.makedirs(target_dir, exist_ok=True)
    file_path = os.path.join(target_dir, filename)
    if not os.path.exists(file_path):
        hf_hub_download(repo_id=repo_id, filename=filename, subfolder=subfolder, local_dir=target_dir)

repo_id = "Comfy-Org/Wan_2.1_ComfyUI_repackaged"
download_model(repo_id, "wan2.1_i2v_720p_14B_fp8_e4m3fn.safetensors", "split_files/diffusion_models", "diffusion_models")
download_model(repo_id, "umt5_xxl_fp8_e4m3fn_scaled.safetensors", "split_files/text_encoders", "text_encoders")
download_model(repo_id, "wan_2.1_vae.safetensors", "split_files/vae", "vae")
download_model(repo_id, "clip_vision_h.safetensors", "split_files/clip_vision", "clip_vision")
download_model(repo_id, "image_to_video_wan_720p_example.json", "example%20workflows_Wan2.1", "user/default/workflows")

# Tải workflow từ foxtonai
workflow_dir = os.path.join(WORK_DIR, "user/default/workflows")
os.makedirs(workflow_dir, exist_ok=True)
subprocess.run(["wget", "-P", workflow_dir, "https://foxtonai.com/downloads/comfyui-wan-2.1-i2v-modified-workflow.json"])