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 = {}