298977887/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2025-04-20 03:13:30 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add original_shape parameter to transformer patch extra_options.

Browse Source
This commit is contained in:
comfyanonymous 2023-06-21 13:22:01 -04:00
parent 852cf4db99
commit 9fccf4aa03
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
comfy/ldm/modules/attention.py
View File

@ -529,6 +529,8 @@ class BasicTransformerBlock(nn.Module):
extra_options["transformer_index"] = transformer_options["current_index"] extra_options["transformer_index"] = transformer_options["current_index"]
if "block_index" in transformer_options: if "block_index" in transformer_options:
extra_options["block_index"] = transformer_options["block_index"] extra_options["block_index"] = transformer_options["block_index"]
if "original_shape" in transformer_options:
extra_options["original_shape"] = transformer_options["original_shape"]
if "patches" in transformer_options: if "patches" in transformer_options:
transformer_patches = transformer_options["patches"] transformer_patches = transformer_options["patches"]
else: else: