Skip to content

Commit 00e1209

Browse files
committed
Add input field.
1 parent 770c12c commit 00e1209

File tree

1 file changed

+7
-0
lines changed

1 file changed

+7
-0
lines changed

src/diffusers/pipelines/lumina/pipeline_lumina.py

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -175,6 +175,13 @@ class LuminaText2ImgPipeline(DiffusionPipeline):
175175

176176
_optional_components = []
177177
model_cpu_offload_seq = "text_encoder->transformer->vae"
178+
_callback_tensor_inputs = [
179+
"latents",
180+
"prompt_embeds",
181+
"add_text_embeds",
182+
"add_time_ids",
183+
]
184+
178185

179186
def __init__(
180187
self,

0 commit comments

Comments
 (0)