From 5f4fcd19e7a5ce82b998495d18c10f4a111e41b7 Mon Sep 17 00:00:00 2001 From: Jedrzej Kosinski Date: Mon, 30 Mar 2026 07:30:32 -0700 Subject: [PATCH] Simplify multigpu nodes: default max_gpus=2, remove gpu_options input, disable Options node Amp-Thread-ID: https://ampcode.com/threads/T-019d3ee9-19d5-767a-9d7a-e50cbbef815b Co-authored-by: Amp --- comfy_extras/nodes_multigpu.py | 9 ++++----- 1 file changed, 4 insertions(+), 5 deletions(-) diff --git a/comfy_extras/nodes_multigpu.py b/comfy_extras/nodes_multigpu.py index 789038b1d..c77dd5c1f 100644 --- a/comfy_extras/nodes_multigpu.py +++ b/comfy_extras/nodes_multigpu.py @@ -29,8 +29,7 @@ class MultiGPUWorkUnitsNode(io.ComfyNode): description=cleandoc(cls.__doc__), inputs=[ io.Model.Input("model"), - io.Int.Input("max_gpus", default=8, min=1, step=1), - io.Custom("GPU_OPTIONS").Input("gpu_options", optional=True), + io.Int.Input("max_gpus", default=2, min=1, step=1), ], outputs=[ io.Model.Output(), @@ -38,8 +37,8 @@ class MultiGPUWorkUnitsNode(io.ComfyNode): ) @classmethod - def execute(cls, model: ModelPatcher, max_gpus: int, gpu_options: comfy.multigpu.GPUOptionsGroup = None) -> io.NodeOutput: - model = comfy.multigpu.create_multigpu_deepclones(model, max_gpus, gpu_options, reuse_loaded=True) + def execute(cls, model: ModelPatcher, max_gpus: int) -> io.NodeOutput: + model = comfy.multigpu.create_multigpu_deepclones(model, max_gpus, reuse_loaded=True) return io.NodeOutput(model) @@ -82,7 +81,7 @@ class MultiGPUExtension(ComfyExtension): async def get_node_list(self) -> list[type[io.ComfyNode]]: return [ MultiGPUWorkUnitsNode, - MultiGPUOptionsNode, + # MultiGPUOptionsNode, ]