ltx_video.yml modified

prishajain1 · prishajain1 · commit 12886226d777 · 2026-01-27T11:05:17.000+05:30
diff --git a/src/maxdiffusion/configs/ltx_video.yml b/src/maxdiffusion/configs/ltx_video.yml
@@ -1,6 +1,7 @@
 #hardware
 hardware: 'tpu'
 skip_jax_distributed_system: False
+attention: 'flash'
 
 jax_cache_dir: ''
 weights_dtype: 'bfloat16'
