Update app.py

app.py

@@ -11,55 +11,27 @@ if not os.path.exists(WORK_DIR):
     os.makedirs(WORK_DIR)
     subprocess.run(["git", "clone", "--depth", "1", "https://github.com/comfyanonymous/ComfyUI.git", WORK_DIR])
 
-# Tải và cài đặt custom nodes (từ Dockerfile của bạn)
-CUSTOM_NODES_DIR = os.path.join(WORK_DIR, "custom_nodes")
-os.makedirs(CUSTOM_NODES_DIR, exist_ok=True)
-custom_nodes = [
-    "https://github.com/ltdrdata/ComfyUI-Manager.git",
-    "https://github.com/Kosinkadink/ComfyUI-VideoHelperSuite.git",
-    "https://github.com/FizzleDorf/ComfyUI_FizzNodes.git",
-    "https://github.com/Stability-AI/stability-ComfyUI-nodes.git",
-    "https://github.com/EllangoK/ComfyUI-post-processing-nodes.git",
-    "https://github.com/TinyTerra/ComfyUI_tinyterraNodes.git",
-    "https://github.com/WASasquatch/was-node-suite-comfyui.git",
-    "https://github.com/WASasquatch/PowerNoiseSuite.git",
-    "https://github.com/Kosinkadink/ComfyUI-AnimateDiff-Evolved.git"
-]
-for repo in custom_nodes:
-    repo_name = repo.split("/")[-1].replace(".git", "")
-    target_dir = os.path.join(CUSTOM_NODES_DIR, repo_name)
-    if not os.path.exists(target_dir):
-        subprocess.run(["git", "clone", "--depth", "1", repo, target_dir])
-
-# Tải mô hình AnimateDiff
-animate_diff_dir = os.path.join(CUSTOM_NODES_DIR, "ComfyUI-AnimateDiff-Evolved/models")
-os.makedirs(animate_diff_dir, exist_ok=True)
-for model in ["mm_sd_v15_v2.ckpt", "mm_sd_v15.ckpt"]:
-    model_path = os.path.join(animate_diff_dir, model)
-    if not os.path.exists(model_path):
-        subprocess.run(["wget", "-c", f"https://huggingface.co/guoyww/animatediff/resolve/main/{model}", "-P", animate_diff_dir])
-
-# Tải mô hình WAN 2.1 I2V-14B-720P
+# Tải WAN 2.1 I2V-14B-720P
 MODEL_DIR = os.path.join(WORK_DIR, "models")
 os.makedirs(MODEL_DIR, exist_ok=True)
 
-def download_model(repo_id, filename, subfolder):
-    target_dir = os.path.join(MODEL_DIR, subfolder)
+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, local_dir=target_dir)
+        hf_hub_download(repo_id=repo_id, filename=filename, subfolder=subfolder, local_dir=target_dir)
         print(f"Đã tải {filename} vào {target_dir}")
 
 repo_id = "Comfy-Org/Wan_2.1_ComfyUI_repackaged"
-download_model(repo_id, "wan2.1_i2v_720p_14B_fp8_e4m3fn.safetensors", "diffusion_models")
-download_model(repo_id, "wan_2.1_vae.safetensors", "vae")
-download_model(repo_id, "umt5_xxl_fp8_e4m3fn_scaled.safetensors", "text_encoders")
-download_model(repo_id, "clip_vision_h.safetensors", "clip_vision")
+download_model(repo_id, "wan2.1_i2v_720p_14B_fp8_e4m3fn.safetensors", "split_files/diffusion_models", "diffusion_models")
+download_model(repo_id, "wan_2.1_vae.safetensors", "split_files/vae", "vae")
+download_model(repo_id, "umt5_xxl_fp8_e4m3fn_scaled.safetensors", "split_files/text_encoders", "text_encoders")
+download_model(repo_id, "clip_vision_h.safetensors", "split_files/clip_vision", "clip_vision")
 
 # Khởi động ComfyUI
 def start_comfyui():
-    subprocess.Popen(["python", "main.py", "--port", "7860", "--listen", "0.0.0.0", "--use-split-cross-attention"], cwd=WORK_DIR)
+    subprocess.Popen(["python", "main.py", "--port", "7860", "--listen", "0.0.0.0"], cwd=WORK_DIR)
     time.sleep(5)
 
 start_comfyui()
@@ -73,8 +45,7 @@ demo = gr.Interface(
     fn=lambda x: "Dùng giao diện ComfyUI bên dưới",
     inputs=gr.Textbox(label="Thông tin", value="ComfyUI với WAN 2.1 I2V-14B-720P"),
     outputs=gr.Textbox(label="Trạng thái"),
-    title="ComfyUI Advanced - WAN 2.1 I2V",
-    description="Chạy workflow Image-to-Video 720P.",
+    title="ComfyUI WAN 2.1 I2V",
     additional_inputs=[HTML(comfyui_interface)]
 )