wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-05-27 17:37:39 +08:00
Code Issues Actions 17 Packages Projects Releases Wiki Activity

multigpu: trim _force_supported_compute_dtype docstring to one line

Browse Source 
Amp-Thread-ID: https://ampcode.com/threads/T-019e61db-ffb1-73a6-b2a8-3d23d7b05792
Co-authored-by: Amp <amp@ampcode.com>
This commit is contained in:
Jedrzej Kosinski 2026-05-25 19:58:36 -07:00
parent 8969bbbf02
commit f663018950
1 changed files with 1 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
comfy_extras/nodes_multigpu.py
View File

@ -49,11 +49,7 @@ class MultiGPUCFGSplitNode(io.ComfyNode):
def _force_supported_compute_dtype(patcher: ModelPatcher, device: torch.device): def _force_supported_compute_dtype(patcher: ModelPatcher, device: torch.device):
"""Cast compute dtype to one the device supports; no-op if already supported. """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)."""
weight_dtype = patcher.model_dtype() weight_dtype = patcher.model_dtype()
cast_dtype = comfy.model_management.unet_manual_cast(weight_dtype, device) cast_dtype = comfy.model_management.unet_manual_cast(weight_dtype, device)
if cast_dtype is None: if cast_dtype is None: