Fix regression from last commit.

39e75862 · comfyanonymous · 50dc39d6 · 39e75862
Commit 39e75862 authored Nov 26, 2023 by comfyanonymous
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

comfy/ldm/modules/attention.py comfy/ldm/modules/attention.py +2 -2

No files found.
--- a/comfy/ldm/modules/attention.py
+++ b/comfy/ldm/modules/attention.py
@@ -428,8 +428,8 @@ class BasicTransformerBlock(nn.Module):

    def _forward(self, x, context=None, transformer_options={}):
        extra_options = {}
-        block = None
-        block_index = 0
+        block = transformer_options.get("block", None)
+        block_index = transformer_options.get("block_index", 0)
        transformer_patches = {}
        transformer_patches_replace = {}