ltx_video.yml modified

prishajain1 · prishajain1 · commit ec59a6f656de · 2026-01-27T11:07:36.000+05:30
diff --git a/src/maxdiffusion/configs/ltx_video.yml b/src/maxdiffusion/configs/ltx_video.yml
@@ -2,6 +2,7 @@
 hardware: 'tpu'
 skip_jax_distributed_system: False
 attention: 'flash'
+attention_sharding_uniform: True 
 
 jax_cache_dir: ''
 weights_dtype: 'bfloat16'
