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

Merge b73dd32462 into b08debceca

Browse Source
This commit is contained in:
chinesewebman 2026-07-06 11:48:11 +09:00 committed by GitHub
commit ccae3939d6
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
comfy/ops.py
View File

@ -317,7 +317,10 @@ def cast_bias_weight(s, input=None, dtype=None, device=None, bias_dtype=None, of
prefetched = hasattr(s, "_prefetch")
offload_stream = None
offload_device = None
if not prefetched:
# comfy_aimdo is only registered for CUDA / ROCm backends. On MPS (and
# CPU) we have to skip the vbar fast path and fall back to the plain
# cast in resolve_cast_module_with_vbar below.
if not prefetched and device.type not in ("cpu", "mps"):
offload_stream = cast_modules_with_vbar([s], dtype, device, bias_dtype, non_blocking)
comfy.model_management.sync_stream(device, offload_stream)