comfyanonymous/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-04-28 10:21:27 +00:00
Code Issues Packages Projects Releases Wiki Activity

Make WeightHooks compatible with MultiGPU, clean up some code

Browse Source
This commit is contained in:
Jedrzej Kosinski
2025-01-20 04:34:26 -06:00
parent bfce723311
commit 328d4f16a9
4 changed files with 86 additions and 26 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
comfy_extras/nodes_multigpu.py
View File

@@ -28,7 +28,7 @@ class MultiGPUInitialize:
model = model.clone()
comfy.model_management.unload_all_models()
for device in extra_devices:
device_patcher = model.multigpu_clone(new_load_device=device)
device_patcher = model.multigpu_deepclone(new_load_device=device)
device_patcher.is_multigpu_clone = True
multigpu_models = model.get_additional_models_with_key("multigpu")
multigpu_models.append(device_patcher)