nodes/pollockjj-ComfyUI-MultiGPU
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Merge pull request #12 from 3dluvr/main

Browse Source 
Fix check_module_exists() to use folder_paths
This commit is contained in:
John Pollock
2025-01-29 05:12:49 -06:00
committed by GitHub
parent 41e974f086 379ecce687
commit fc4e2b2f12
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
__init__.py
+2 -2
View File
@@ -369,7 +369,7 @@ def override_class_with_distorch(cls):
NODE_CLASS_MAPPINGS = {"DeviceSelectorMultiGPU": DeviceSelectorMultiGPU}
def check_module_exists(module_path):
full_path = os.path.join("custom_nodes", module_path)
full_path = os.path.join(folder_paths.get_folder_paths("custom_nodes")[0], module_path)
logging.info(f"MultiGPU: Checking for module at {full_path}")
if not os.path.exists(full_path):
@@ -1048,4 +1048,4 @@ if check_module_exists("ComfyUI-HunyuanVideoWrapper") or check_module_exists("co
register_DownloadAndLoadHyVideoTextEncoder()
logging.info(f"MultiGPU: Registration complete. Final mappings: {', '.join(NODE_CLASS_MAPPINGS.keys())}")
logging.info(f"MultiGPU: Registration complete. Final mappings: {', '.join(NODE_CLASS_MAPPINGS.keys())}")