Initial proof of concept of giving splitting cond sampling between multiple GPUs

This commit is contained in:
Jedrzej Kosinski
2025-01-08 03:33:05 -06:00
parent 871258aa72
commit 7448f02b7c
3 changed files with 198 additions and 4 deletions

View File

@@ -218,6 +218,8 @@ class ModelPatcher:
self.is_clip = False
self.hook_mode = comfy.hooks.EnumHookMode.MaxSpeed
self.is_multigpu_clone = False
if not hasattr(self.model, 'model_loaded_weight_memory'):
self.model.model_loaded_weight_memory = 0
@@ -293,6 +295,8 @@ class ModelPatcher:
n.is_clip = self.is_clip
n.hook_mode = self.hook_mode
n.is_multigpu_clone = self.is_multigpu_clone
for callback in self.get_all_callbacks(CallbacksMP.ON_CLONE):
callback(self, n)
return n