diff --git a/extension-node-map.json b/extension-node-map.json index e0e4b73..ef5257c 100644 --- a/extension-node-map.json +++ b/extension-node-map.json @@ -401,6 +401,7 @@ "CLIP Vision Input Switch", "CLIPSeg Batch Masking", "CLIPSeg Masking", + "CLIPTextEncode (BlenderNeko Advanced + NSP)", "CLIPTextEncode (NSP)", "Cache Node", "Checkpoint Loader", @@ -603,6 +604,10 @@ "Single Text String", "String to ASCII" ], + "https://github.com/city96/ComfyUI_NetDist": [ + "FetchRemote", + "QueueRemote" + ], "https://github.com/comfyanonymous/ComfyUI_experiments/raw/master/sampler_rescalecfg.py": [ "RescaleClassifierFreeGuidanceTest" ], diff --git a/scanner.py b/scanner.py index 2866d28..88e4288 100644 --- a/scanner.py +++ b/scanner.py @@ -26,10 +26,18 @@ def scan_in_file(filename): for key, value in class_dict.items(): nodes.add(key) + + update_pattern = r"NODE_CLASS_MAPPINGS.update\s*\({([^}]*)}\)" + update_match = re.search(update_pattern, code) + if update_match: + update_dict_text = update_match.group(1) + update_key_value_pairs = re.findall(r"\"([^\"]*)\"\s*:\s*([^,\n]*)", update_dict_text) + for key, value in update_key_value_pairs: + class_dict[key] = value.strip() + nodes.add(key) return nodes - def get_py_file_paths(dirname): file_paths = []