20
20
model_loaders_info = json .load (open (os .path .join (os .path .dirname (__file__ ), "model_loader_info.json" )))
21
21
node_deps_info = json .load (open (os .path .join (os .path .dirname (__file__ ), "node_deps_info.json" )))
22
22
node_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" )))
23
24
24
25
model_suffix = [".ckpt" , ".safetensors" , ".bin" , ".pth" , ".pt" , ".onnx" , ".gguf" , ".sft" ]
25
26
extra_packages = ["transformers" , "timm" , "diffusers" , "accelerate" ]
@@ -184,9 +185,17 @@ def collect_unknown_models(filename):
184
185
if node_class_type is None :
185
186
raise NotImplementedError (f"Missing nodes founded, please first install the missing nodes using ComfyUI Manager" )
186
187
node_cls = NODE_CLASS_MAPPINGS [node_class_type ]
188
+
189
+ skip_model_check = False
190
+
187
191
if hasattr (node_cls , "RELATIVE_PYTHON_MODULE" ) and node_cls .RELATIVE_PYTHON_MODULE .startswith ("custom_nodes." ):
188
192
print (node_cls .RELATIVE_PYTHON_MODULE )
189
193
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
+
190
199
if node_class_type in model_loaders_info :
191
200
for field_name , filename in node_info ["inputs" ].items ():
192
201
if type (filename ) != str :
@@ -204,7 +213,7 @@ def collect_unknown_models(filename):
204
213
"filename" : filename ,
205
214
"rel_save_path" : rel_save_path
206
215
}
207
- else :
216
+ elif not skip_model_check :
208
217
tree_map (collect_unknown_models , node_info ["inputs" ])
209
218
210
219
list (map (partial (collect_local_file , mapping_dict = file_mapping_dict ), node_info ["inputs" ].values ()))
0 commit comments