fix

prishajain1 · prishajain1 · commit 832a43dcc19a · 2026-02-27T22:13:36.000+05:30
diff --git a/src/maxdiffusion/pipelines/ltx2/ltx2_pipeline.py b/src/maxdiffusion/pipelines/ltx2/ltx2_pipeline.py
@@ -1121,8 +1121,8 @@ def __call__(
       )
 
       # 3. Prepare latents
-      # Effective batch size
-      batch_size = prompt_embeds.shape[0] // 2 if guidance_scale > 1.0 else prompt_embeds.shape[0]
+      _bs = prompt_embeds[0].shape[0] if isinstance(prompt_embeds, list) else prompt_embeds.shape[0]
+      batch_size = _bs // 2 if guidance_scale > 1.0 else _bs
       
       # Prepare generators
       if generator is None:

Original file line number	Diff line number	Diff line change
`@@ -1121,8 +1121,8 @@ def __call__(`
`1121`	`1121`	`)`
`1122`	`1122`
`1123`	`1123`	`# 3. Prepare latents`
`1124`		`- # Effective batch size`
`1125`		`- batch_size = prompt_embeds.shape[0] // 2 if guidance_scale > 1.0 else prompt_embeds.shape[0]`
	`1124`	`+ _bs = prompt_embeds[0].shape[0] if isinstance(prompt_embeds, list) else prompt_embeds.shape[0]`
	`1125`	`+ batch_size = _bs // 2 if guidance_scale > 1.0 else _bs`
`1126`	`1126`
`1127`	`1127`	`# Prepare generators`
`1128`	`1128`	`if generator is None:`