pipeline output class added

Author: prishajain1

src/maxdiffusion/pipelines/ltx2/ltx2_pipeline.py

@@ -28,9 +28,6 @@
 from transformers import AutoTokenizer, GemmaTokenizer, GemmaTokenizerFast, Gemma3ForConditionalGeneration
 from tqdm.auto import tqdm
 import qwix
-
-from ...utils import logging
-
 from ...utils import logging
 from ...schedulers import FlaxFlowMatchScheduler
 from ...models.ltx2.autoencoder_kl_ltx2 import LTX2VideoAutoencoderKL
@@ -46,6 +43,11 @@
 from ... import max_utils
 from ...max_utils import get_flash_block_sizes, get_precision, device_put_replicated
 
+@flax.struct.dataclass
+class LTX2PipelineOutput:
+    frames: jax.Array
+    audio: Optional[jax.Array] = None
+
 logger = logging.get_logger(__name__)