From 39e75862b248a20e8233ccee743ba5b2e977cdcf Mon Sep 17 00:00:00 2001
From: comfyanonymous <comfyanonymous@protonmail.com>
Date: Sun, 26 Nov 2023 03:43:02 -0500
Subject: [PATCH] Fix regression from last commit.

---
 comfy/ldm/modules/attention.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/comfy/ldm/modules/attention.py b/comfy/ldm/modules/attention.py
index 7dc1a1b5..f6845238 100644
--- 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 = {}