adding support for use_batched_guidance flag

Author: prishajain1

src/maxdiffusion/generate_ltx2.py

@@ -106,6 +106,7 @@ def call_pipeline(config, pipeline, prompt, negative_prompt):
       modality_scale=getattr(config, "modality_scale", 1.0),
       audio_modality_scale=getattr(config, "audio_modality_scale", None),
       use_cross_timestep=getattr(config, "use_cross_timestep", None),
+      use_batched_guidance=getattr(config, "use_batched_guidance", True),
       noise_scale=getattr(config, "noise_scale", 1.0),
       dtype=jnp.bfloat16 if getattr(config, "activations_dtype", "bfloat16") == "bfloat16" else jnp.float32,
   )