Skip to content

Commit 08fec26

Browse files
committed
reverted VAE compression changes
1 parent 8641526 commit 08fec26

1 file changed

Lines changed: 2 additions & 9 deletions

File tree

src/maxdiffusion/pipelines/ltx2/ltx2_pipeline.py

Lines changed: 2 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -235,15 +235,8 @@ def __init__(
235235
self.vae_temporal_compression_ratio = getattr(self.vae, "temporal_compression_ratio", 8)
236236

237237
# Audio VAE compression ratios
238-
if hasattr(self.audio_vae, "config") and hasattr(self.audio_vae.config, "patch_size"):
239-
self.audio_vae_mel_compression_ratio = self.audio_vae.config.patch_size
240-
else:
241-
self.audio_vae_mel_compression_ratio = getattr(self.audio_vae, "mel_compression_ratio", 1)
242-
243-
if hasattr(self.audio_vae, "config") and hasattr(self.audio_vae.config, "patch_size_t"):
244-
self.audio_vae_temporal_compression_ratio = self.audio_vae.config.patch_size_t
245-
else:
246-
self.audio_vae_temporal_compression_ratio = getattr(self.audio_vae, "temporal_compression_ratio", 1)
238+
self.audio_vae_mel_compression_ratio = getattr(self.audio_vae, "mel_compression_ratio", 4)
239+
self.audio_vae_temporal_compression_ratio = getattr(self.audio_vae, "temporal_compression_ratio", 4)
247240

248241
# Transformer patch sizes
249242
self.transformer_spatial_patch_size = getattr(self.transformer.config, "patch_size", 1) if getattr(self, "transformer", None) is not None else 1

0 commit comments

Comments
 (0)