From a17f5cb63f0a4fb1843616e82c674490c04a4d6a Mon Sep 17 00:00:00 2001 From: dg845 <58458699+dg845@users.noreply.github.com> Date: Tue, 6 Jan 2026 21:34:57 -0800 Subject: [PATCH] Apply suggestions from code review Co-authored-by: Sayak Paul --- docs/source/en/api/models/autoencoderkl_audio_ltx_2.md | 1 + src/diffusers/models/autoencoders/autoencoder_kl_ltx2_audio.py | 2 +- .../autoencoders/test_models_autoencoder_kl_ltx2_audio.py | 2 +- 3 files changed, 3 insertions(+), 2 deletions(-) diff --git a/docs/source/en/api/models/autoencoderkl_audio_ltx_2.md b/docs/source/en/api/models/autoencoderkl_audio_ltx_2.md index 36d1834e58..d0024474e9 100644 --- a/docs/source/en/api/models/autoencoderkl_audio_ltx_2.md +++ b/docs/source/en/api/models/autoencoderkl_audio_ltx_2.md @@ -24,5 +24,6 @@ vae = AutoencoderKLLTX2Audio.from_pretrained("Lightricks/LTX-2", subfolder="vae" ## AutoencoderKLLTX2Audio [[autodoc]] AutoencoderKLLTX2Audio + - encode - decode - all \ No newline at end of file diff --git a/src/diffusers/models/autoencoders/autoencoder_kl_ltx2_audio.py b/src/diffusers/models/autoencoders/autoencoder_kl_ltx2_audio.py index dc09f44d82..41b543de2f 100644 --- a/src/diffusers/models/autoencoders/autoencoder_kl_ltx2_audio.py +++ b/src/diffusers/models/autoencoders/autoencoder_kl_ltx2_audio.py @@ -668,7 +668,7 @@ class LTX2AudioDecoder(nn.Module): class AutoencoderKLLTX2Audio(ModelMixin, AutoencoderMixin, ConfigMixin): r""" - LTX2 audio VAE. Currently, only implements the decoder. + LTX2 audio VAE for encoding and decoding audio latent representations. """ _supports_gradient_checkpointing = False diff --git a/tests/models/autoencoders/test_models_autoencoder_kl_ltx2_audio.py b/tests/models/autoencoders/test_models_autoencoder_kl_ltx2_audio.py index 3c10330e20..ce93dfb42a 100644 --- a/tests/models/autoencoders/test_models_autoencoder_kl_ltx2_audio.py +++ b/tests/models/autoencoders/test_models_autoencoder_kl_ltx2_audio.py @@ -83,6 +83,6 @@ class AutoencoderKLLTX2AudioTests(ModelTesterMixin, AutoencoderTesterMixin, unit def test_outputs_equivalence(self): pass - @unittest.skip("AutoencoderKLLTXAudio does not support `norm_num_groups` because it does not use GroupNorm.") + @unittest.skip("AutoencoderKLLTX2Audio does not support `norm_num_groups` because it does not use GroupNorm.") def test_forward_with_norm_groups(self): pass