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