From 9fccf4aa031a3e0698c03bd5424c78e56fba9f08 Mon Sep 17 00:00:00 2001
From: comfyanonymous <comfyanonymous@protonmail.com>
Date: Wed, 21 Jun 2023 13:22:01 -0400
Subject: [PATCH] Add original_shape parameter to transformer patch
 extra_options.

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

diff --git a/comfy/ldm/modules/attention.py b/comfy/ldm/modules/attention.py
index a0d69569..68a4ef6e 100644
--- a/comfy/ldm/modules/attention.py
+++ b/comfy/ldm/modules/attention.py
@@ -529,6 +529,8 @@ class BasicTransformerBlock(nn.Module):
             extra_options["transformer_index"] = transformer_options["current_index"]
         if "block_index" in transformer_options:
             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:
             transformer_patches = transformer_options["patches"]
         else: