From ffbecfffb953914f5b4bd8f61d810ff2300631de Mon Sep 17 00:00:00 2001
From: comfyanonymous <121283862+comfyanonymous@users.noreply.github.com>
Date: Tue, 7 Jul 2026 21:17:31 -0700
Subject: [PATCH] Fix crash when using UNetSelfAttentionMultiply (#14823)

---
 comfy/ops.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/comfy/ops.py b/comfy/ops.py
index 69d32e254..35a1ee31e 100644
--- a/comfy/ops.py
+++ b/comfy/ops.py
@@ -174,6 +174,8 @@ def cast_modules_with_vbar(comfy_modules, dtype, device, bias_dtype, non_blockin
                     elif xfer_dest2 is not None:
                         xfer_source.prepare(xfer_dest2, stream, copy=True, commit=False)
                         return
+                    else:
+                        return
                 comfy.model_management.cast_to_gathered(xfer_source, xfer_dest, non_blocking=non_blocking, stream=stream, r2=xfer_dest2)
 
         def handle_pin(m, pin, source, dest, subset="weights", size=None):