From f663018950e25bb71ed2023699359fe4cc664a17 Mon Sep 17 00:00:00 2001 From: Jedrzej Kosinski Date: Mon, 25 May 2026 19:58:36 -0700 Subject: [PATCH] multigpu: trim _force_supported_compute_dtype docstring to one line Amp-Thread-ID: https://ampcode.com/threads/T-019e61db-ffb1-73a6-b2a8-3d23d7b05792 Co-authored-by: Amp --- comfy_extras/nodes_multigpu.py | 6 +----- 1 file changed, 1 insertion(+), 5 deletions(-) diff --git a/comfy_extras/nodes_multigpu.py b/comfy_extras/nodes_multigpu.py index 878d85baf..d2f6fe67a 100644 --- a/comfy_extras/nodes_multigpu.py +++ b/comfy_extras/nodes_multigpu.py @@ -49,11 +49,7 @@ class MultiGPUCFGSplitNode(io.ComfyNode): def _force_supported_compute_dtype(patcher: ModelPatcher, device: torch.device): - """Cast compute dtype to one the device supports; no-op if already supported. - - Uses unet_manual_cast which encodes per-device dtype support (e.g. CPU - rejects fp16/bf16, falling back to fp32 to avoid PyTorch's ~500-600x - slower software emulation).""" + """Cast compute dtype to one the device supports; no-op if already supported.""" weight_dtype = patcher.model_dtype() cast_dtype = comfy.model_management.unet_manual_cast(weight_dtype, device) if cast_dtype is None: