remat policy = None

Author: prishajain1

src/maxdiffusion/configs/ltx2_video.yml

@@ -4,7 +4,7 @@ skip_jax_distributed_system: False
 attention: 'flash'
 attention_sharding_uniform: True 
 precision: 'bf16'
-remat_policy: "full_save"
+remat_policy: "NONE"
 names_which_can_be_saved: []
 names_which_can_be_offloaded: []