Skip to content

Commit ffbecff

Browse files
Fix crash when using UNetSelfAttentionMultiply (Comfy-Org#14823)
1 parent b481bc1 commit ffbecff

1 file changed

Lines changed: 2 additions & 0 deletions

File tree

comfy/ops.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -174,6 +174,8 @@ def cast_maybe_lowvram_patch(xfer_source, xfer_dest, stream, xfer_dest2=None):
174174
elif xfer_dest2 is not None:
175175
xfer_source.prepare(xfer_dest2, stream, copy=True, commit=False)
176176
return
177+
else:
178+
return
177179
comfy.model_management.cast_to_gathered(xfer_source, xfer_dest, non_blocking=non_blocking, stream=stream, r2=xfer_dest2)
178180

179181
def handle_pin(m, pin, source, dest, subset="weights", size=None):

0 commit comments

Comments
 (0)