From d2573428f5d307f4d308820c1e364f50dfbee4fa Mon Sep 17 00:00:00 2001
From: wl-zhao <zhaowl20@mails.tsinghua.edu.cn>
Date: Sat, 12 Oct 2024 13:33:21 +0800
Subject: [PATCH] update

---
 comfy-nodes/output_image.py |  7 ++-----
 dependency_checker.py       | 38 ++++++++++++++++++++-----------------
 2 files changed, 23 insertions(+), 22 deletions(-)

diff --git a/comfy-nodes/output_image.py b/comfy-nodes/output_image.py
index 800a310..3198e4f 100644
--- a/comfy-nodes/output_image.py
+++ b/comfy-nodes/output_image.py
@@ -69,11 +69,8 @@ class ShellAgentSaveVideoVHS:
     def validate(cls, **kwargs):
         schema = {
             "title": kwargs["output_name"],
-            "type": "array",
-            "items": {
-                "type": "string",
-                "url_type": "video",
-            }
+            "type": "string",
+            "url_type": "video",
         }
         return schema
         
diff --git a/dependency_checker.py b/dependency_checker.py
index 9e05cd0..092ddc3 100644
--- a/dependency_checker.py
+++ b/dependency_checker.py
@@ -23,26 +23,30 @@ ComfyUIModelLoaders = {
     'CLIPVisionLoader': (["clip_name"], "clip_vision"),
     'StyleModelLoader': (["style_model_name"], "style_models"),
     'GLIGENLoader': (["gligen_name"], "gligen"),
+    'ImageOnlyCheckpointLoader': (["ckpt_name"], "checkpoints"),
+    "UpscaleModelLoader": (["model_name"], "upscale_models"),
+    "TripleCLIPLoader": (["clip_name1", "clip_name2", "clip_name3"], "clip"),
+    "HypernetworkLoader": (["hypernetwork_name"], "hypernetworks")
 }
 
 
-ComfyUIFileLoaders = {
-    'VAELoader': (["vae_name"], "vae"),
-    'CheckpointLoader': (["ckpt_name"], "checkpoints"),
-    'CheckpointLoaderSimple': (["ckpt_name"], "checkpoints"),
-    'DiffusersLoader': (["model_path"], "diffusers"),
-    'unCLIPCheckpointLoader': (["ckpt_name"], "checkpoints"),
-    'LoraLoader': (["lora_name"], "loras"),
-    'LoraLoaderModelOnly': (["lora_name"], "loras"),
-    'ControlNetLoader': (["control_net_name"], "controlnet"),
-    'DiffControlNetLoader': (["control_net_name"], "controlnet"),
-    'UNETLoader': (["unet_name"], "unet"),
-    'CLIPLoader': (["clip_name"], "clip"),
-    'DualCLIPLoader': (["clip_name1", "clip_name2"], "clip"),
-    'CLIPVisionLoader': (["clip_name"], "clip_vision"),
-    'StyleModelLoader': (["style_model_name"], "style_models"),
-    'GLIGENLoader': (["gligen_name"], "gligen"),
-}
+# ComfyUIFileLoaders = {
+#     'VAELoader': (["vae_name"], "vae"),
+#     'CheckpointLoader': (["ckpt_name"], "checkpoints"),
+#     'CheckpointLoaderSimple': (["ckpt_name"], "checkpoints"),
+#     'DiffusersLoader': (["model_path"], "diffusers"),
+#     'unCLIPCheckpointLoader': (["ckpt_name"], "checkpoints"),
+#     'LoraLoader': (["lora_name"], "loras"),
+#     'LoraLoaderModelOnly': (["lora_name"], "loras"),
+#     'ControlNetLoader': (["control_net_name"], "controlnet"),
+#     'DiffControlNetLoader': (["control_net_name"], "controlnet"),
+#     'UNETLoader': (["unet_name"], "unet"),
+#     'CLIPLoader': (["clip_name"], "clip"),
+#     'DualCLIPLoader': (["clip_name1", "clip_name2"], "clip"),
+#     'CLIPVisionLoader': (["clip_name"], "clip_vision"),
+#     'StyleModelLoader': (["style_model_name"], "style_models"),
+#     'GLIGENLoader': (["gligen_name"], "gligen"),
+# }
 
 
 model_list_json = json.load(open(os.path.join(os.path.dirname(__file__), "model_info.json")))