seed added to config

Author: prishajain1

src/maxdiffusion/configs/ltx2_video.yml

@@ -90,3 +90,4 @@ bwd_quantization_calibration_method: "absmax"
 qwix_module_path: ".*"
 jit_initializers: True 
 enable_single_replica_ckpt_restoring: False
+seed: 0

src/maxdiffusion/pipelines/ltx2/ltx2_pipeline.py

@@ -1010,7 +1010,8 @@ def prepare_latents(
 
     shape = (batch_size, num_channels_latents, num_frames, height, width)
     if generator is None:
-      generator = jax.random.key(0)
+      seed = getattr(self.config, "seed", 1) if hasattr(self, "config") else 1
+      generator = jax.random.key(seed)
 
     latents = jax.random.normal(generator, shape, dtype=dtype or jnp.float32)
     latents = self._pack_latents(latents, self.transformer_spatial_patch_size, self.transformer_temporal_patch_size)