Added unload_model_and_clones to model_management.py to allow unloading only relevant models

2025-01-23 01:20:00 -06:00
parent ef137ac0b6
commit 02a4d0ad7d
4 changed files with 14 additions and 3 deletions
--- a/comfy_extras/nodes_multigpu.py
+++ b/comfy_extras/nodes_multigpu.py
@@ -26,7 +26,7 @@ class MultiGPUInitialize:
        extra_devices = extra_devices[:max_gpus-1]
        if len(extra_devices) > 0:
            model = model.clone()
-            comfy.model_management.unload_all_models()
+            comfy.model_management.unload_model_and_clones(model)
            for device in extra_devices:
                device_patcher = model.multigpu_deepclone(new_load_device=device)
                device_patcher.is_multigpu_clone = True