From 407a5a656f103c42497f5938a80d0771712b8613 Mon Sep 17 00:00:00 2001 From: Jedrzej Kosinski Date: Fri, 28 Mar 2025 02:48:11 -0500 Subject: [PATCH] Rollback core of last commit due to weird behavior --- comfy/multigpu.py | 14 +++++++------- 1 file changed, 7 insertions(+), 7 deletions(-) diff --git a/comfy/multigpu.py b/comfy/multigpu.py index 26edcee9..90995a5a 100644 --- a/comfy/multigpu.py +++ b/comfy/multigpu.py @@ -99,13 +99,13 @@ def create_multigpu_deepclones(model: ModelPatcher, max_gpus: int, gpu_options: gpu_options.register(model) else: logging.info("No extra torch devices need initialization, skipping initializing MultiGPU Work Units.") - # only keep model clones that don't go 'past' the intended max_gpu count - multigpu_models = model.get_additional_models_with_key("multigpu") - new_multigpu_models = [] - for m in multigpu_models: - if m.load_device in limit_extra_devices: - new_multigpu_models.append(m) - model.set_additional_models("multigpu", new_multigpu_models) + # TODO: only keep model clones that don't go 'past' the intended max_gpu count + # multigpu_models = model.get_additional_models_with_key("multigpu") + # new_multigpu_models = [] + # for m in multigpu_models: + # if m.load_device in limit_extra_devices: + # new_multigpu_models.append(m) + # model.set_additional_models("multigpu", new_multigpu_models) # persist skip_devices for use in sampling code # if len(skip_devices) > 0 or "multigpu_skip_devices" in model.model_options: # model.model_options["multigpu_skip_devices"] = skip_devices