spatial and temporal compression ratio overrides

prishajain1 · prishajain1 · commit c7b5ac4e4389 · 2026-04-11T11:10:22.000+05:30
diff --git a/src/maxdiffusion/pipelines/ltx2/ltx2_pipeline.py b/src/maxdiffusion/pipelines/ltx2/ltx2_pipeline.py
@@ -455,6 +455,8 @@ def create_model(rngs: nnx.Rngs, config: HyperParameters):
                 "decoder_causal": False,
                 "encoder_spatial_padding_mode": "zeros",
                 "decoder_spatial_padding_mode": "zeros",
+                "spatial_compression_ratio": 32,
+                "temporal_compression_ratio": 8,
             }
         )
         vae = LTX2VideoAutoencoderKL(

Original file line number	Diff line number	Diff line change
`@@ -455,6 +455,8 @@ def create_model(rngs: nnx.Rngs, config: HyperParameters):`
`455`	`455`	`"decoder_causal": False,`
`456`	`456`	`"encoder_spatial_padding_mode": "zeros",`
`457`	`457`	`"decoder_spatial_padding_mode": "zeros",`
	`458`	`+ "spatial_compression_ratio": 32,`
	`459`	`+ "temporal_compression_ratio": 8,`
`458`	`460`	`}`
`459`	`461`	`)`
`460`	`462`	`vae = LTX2VideoAutoencoderKL(`