Apply suggestions from code review-2

ParagEkbote · hlky · web-flow · commit 154f34d7dbf8 · 2025-02-13T20:31:56.000+05:30
Co-authored-by: hlky &lt;hlky@hlky.ac&gt;
diff --git a/src/diffusers/pipelines/aura_flow/pipeline_aura_flow.py b/src/diffusers/pipelines/aura_flow/pipeline_aura_flow.py
@@ -135,8 +135,6 @@ class AuraFlowPipeline(DiffusionPipeline):
     _callback_tensor_inputs = [
         "latents",
         "prompt_embeds",
-        "add_text_embeds",
-        "add_time_ids",
     ]
 
     def __init__(
diff --git a/src/diffusers/pipelines/lumina/pipeline_lumina.py b/src/diffusers/pipelines/lumina/pipeline_lumina.py
@@ -178,8 +178,6 @@ class LuminaText2ImgPipeline(DiffusionPipeline):
     _callback_tensor_inputs = [
         "latents",
         "prompt_embeds",
-        "add_text_embeds",
-        "add_time_ids",
     ]
 
 

Original file line number	Diff line number	Diff line change
`@@ -135,8 +135,6 @@ class AuraFlowPipeline(DiffusionPipeline):`
`135`	`135`	`_callback_tensor_inputs = [`
`136`	`136`	`"latents",`
`137`	`137`	`"prompt_embeds",`
`138`		`- "add_text_embeds",`
`139`		`- "add_time_ids",`
`140`	`138`	`]`
`141`	`139`
`142`	`140`	`def __init__(`
Original file line number	Diff line number	Diff line change
`@@ -178,8 +178,6 @@ class LuminaText2ImgPipeline(DiffusionPipeline):`
`178`	`178`	`_callback_tensor_inputs = [`
`179`	`179`	`"latents",`
`180`	`180`	`"prompt_embeds",`
`181`		`- "add_text_embeds",`
`182`		`- "add_time_ids",`
`183`	`181`	`]`
`184`	`182`
`185`	`183`