Fix regression from last commit.

This commit is contained in:
comfyanonymous 2023-11-26 03:43:02 -05:00
parent 50dc39d6ec
commit 39e75862b2

View File

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