From ffa4123f07986da79f27ccf3057880cc002da620 Mon Sep 17 00:00:00 2001 From: wl-zhao Date: Tue, 12 Nov 2024 16:58:27 +0800 Subject: [PATCH] support skip model check for some nodes --- dependency_checker.py | 11 ++++++++++- node_remote.json | 3 +++ 2 files changed, 13 insertions(+), 1 deletion(-) create mode 100644 node_remote.json diff --git a/dependency_checker.py b/dependency_checker.py index e4c2843..ecdfe6c 100644 --- a/dependency_checker.py +++ b/dependency_checker.py @@ -20,6 +20,7 @@ model_list_json = json.load(open(os.path.join(os.path.dirname(__file__), "model_ model_loaders_info = json.load(open(os.path.join(os.path.dirname(__file__), "model_loader_info.json"))) node_deps_info = json.load(open(os.path.join(os.path.dirname(__file__), "node_deps_info.json"))) node_blacklist = json.load(open(os.path.join(os.path.dirname(__file__), "node_blacklist.json"))) +node_remote_skip_models = json.load(open(os.path.join(os.path.dirname(__file__), "node_remote.json"))) model_suffix = [".ckpt", ".safetensors", ".bin", ".pth", ".pt", ".onnx", ".gguf", ".sft"] extra_packages = ["transformers", "timm", "diffusers", "accelerate"] @@ -184,9 +185,17 @@ def resolve_dependencies(prompt, custom_dependencies): # resolve custom nodes an if node_class_type is None: raise NotImplementedError(f"Missing nodes founded, please first install the missing nodes using ComfyUI Manager") node_cls = NODE_CLASS_MAPPINGS[node_class_type] + + skip_model_check = False + if hasattr(node_cls, "RELATIVE_PYTHON_MODULE") and node_cls.RELATIVE_PYTHON_MODULE.startswith("custom_nodes."): print(node_cls.RELATIVE_PYTHON_MODULE) custom_nodes.append(node_cls.RELATIVE_PYTHON_MODULE) + + if node_cls.RELATIVE_PYTHON_MODULE[len("custom_nodes."):] in node_remote_skip_models: + skip_model_check = True + print(f"skip model check for {node_class_type}") + if node_class_type in model_loaders_info: for field_name, filename in node_info["inputs"].items(): if type(filename) != str: @@ -204,7 +213,7 @@ def resolve_dependencies(prompt, custom_dependencies): # resolve custom nodes an "filename": filename, "rel_save_path": rel_save_path } - else: + elif not skip_model_check: tree_map(collect_unknown_models, node_info["inputs"]) list(map(partial(collect_local_file, mapping_dict=file_mapping_dict), node_info["inputs"].values())) diff --git a/node_remote.json b/node_remote.json new file mode 100644 index 0000000..f089f82 --- /dev/null +++ b/node_remote.json @@ -0,0 +1,3 @@ +[ + "BizyAir" +] \ No newline at end of file