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

fix: Ensure model patching fallback logic only applies when 'model_patches_models' attribute is absent.

Browse Source
This commit is contained in:
John Pollock
2026-01-02 22:48:43 -06:00
parent 03e5368f8e
commit 87bde131c4
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
distorch_2.py
+1 -1
View File
@@ -67,7 +67,7 @@ def register_patched_safetensor_modelpatcher():
models_temp.add(m)
model_type = type(m).__name__
if ("GGUF" in model_type or "ModelPatcher" in model_type) and hasattr(m, "model_patches_to"):
if ("GGUF" in model_type or "ModelPatcher" in model_type) and hasattr(m, "model_patches_to") and not hasattr(m, "model_patches_models"):
logger.info(f"[MultiGPU DisTorch V2] {type(m).__name__} missing 'model_patches_models' attribute, using 'model_patches_to' fallback.")
target_device = m.load_device
logger.debug(f"[MultiGPU DisTorch V2] Target device: {target_device}")