AI/diffusers
1
0
Fork 0
mirror of https://github.com/huggingface/diffusers.git synced 2026-01-27 17:22:53 +03:00
Code Activity

Apply suggestions from code review

Browse Source 
Co-authored-by: Sayak Paul <spsayakpaul@gmail.com>
This commit is contained in:
dg845
2026-01-06 21:34:57 -08:00
committed by GitHub
parent 5269ee5067
commit a17f5cb63f
3 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
docs/source/en/api/models/autoencoderkl_audio_ltx_2.md
View File

@@ -24,5 +24,6 @@ vae = AutoencoderKLLTX2Audio.from_pretrained("Lightricks/LTX-2", subfolder="vae"
## AutoencoderKLLTX2Audio
[[autodoc]] AutoencoderKLLTX2Audio
- encode
- decode
- all

2
src/diffusers/models/autoencoders/autoencoder_kl_ltx2_audio.py
View File

@@ -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

2
tests/models/autoencoders/test_models_autoencoder_kl_ltx2_audio.py
View File

@@ -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