1
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2025-08-02 23:14:49 +08:00

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