2020model_loaders_info = json .load (open (os .path .join (os .path .dirname (__file__ ), "model_loader_info.json" )))
2121node_deps_info = json .load (open (os .path .join (os .path .dirname (__file__ ), "node_deps_info.json" )))
2222node_blacklist = json .load (open (os .path .join (os .path .dirname (__file__ ), "node_blacklist.json" )))
23+ node_remote_skip_models = json .load (open (os .path .join (os .path .dirname (__file__ ), "node_remote.json" )))
2324
2425model_suffix = [".ckpt" , ".safetensors" , ".bin" , ".pth" , ".pt" , ".onnx" , ".gguf" , ".sft" ]
2526extra_packages = ["transformers" , "timm" , "diffusers" , "accelerate" ]
@@ -184,9 +185,17 @@ def collect_unknown_models(filename):
184185 if node_class_type is None :
185186 raise NotImplementedError (f"Missing nodes founded, please first install the missing nodes using ComfyUI Manager" )
186187 node_cls = NODE_CLASS_MAPPINGS [node_class_type ]
188+
189+ skip_model_check = False
190+
187191 if hasattr (node_cls , "RELATIVE_PYTHON_MODULE" ) and node_cls .RELATIVE_PYTHON_MODULE .startswith ("custom_nodes." ):
188192 print (node_cls .RELATIVE_PYTHON_MODULE )
189193 custom_nodes .append (node_cls .RELATIVE_PYTHON_MODULE )
194+
195+ if node_cls .RELATIVE_PYTHON_MODULE [len ("custom_nodes." ):] in node_remote_skip_models :
196+ skip_model_check = True
197+ print (f"skip model check for { node_class_type } " )
198+
190199 if node_class_type in model_loaders_info :
191200 for field_name , filename in node_info ["inputs" ].items ():
192201 if type (filename ) != str :
@@ -204,7 +213,7 @@ def collect_unknown_models(filename):
204213 "filename" : filename ,
205214 "rel_save_path" : rel_save_path
206215 }
207- else :
216+ elif not skip_model_check :
208217 tree_map (collect_unknown_models , node_info ["inputs" ])
209218
210219 list (map (partial (collect_local_file , mapping_dict = file_mapping_dict ), node_info ["inputs" ].values ()))
0 commit comments