From 5f7e43d17fe6edf60fe4dcd8b0d8320e84a259ac Mon Sep 17 00:00:00 2001 From: Daniel Gu Date: Tue, 23 Dec 2025 02:08:51 +0100 Subject: [PATCH] Add imports for LTX 2.0 Audio VAE --- src/diffusers/__init__.py | 2 ++ src/diffusers/models/__init__.py | 2 ++ 2 files changed, 4 insertions(+) diff --git a/src/diffusers/__init__.py b/src/diffusers/__init__.py index 71cad3425f..8c6761a07e 100644 --- a/src/diffusers/__init__.py +++ b/src/diffusers/__init__.py @@ -194,6 +194,7 @@ else: "AutoencoderKLHunyuanVideo", "AutoencoderKLHunyuanVideo15", "AutoencoderKLLTXVideo", + "AutoencoderKLLTX2Audio", "AutoencoderKLLTX2Video", "AutoencoderKLMagvit", "AutoencoderKLMochi", @@ -929,6 +930,7 @@ if TYPE_CHECKING or DIFFUSERS_SLOW_IMPORT: AutoencoderKLHunyuanVideo, AutoencoderKLHunyuanVideo15, AutoencoderKLLTXVideo, + AutoencoderKLLTX2Audio, AutoencoderKLLTX2Video, AutoencoderKLMagvit, AutoencoderKLMochi, diff --git a/src/diffusers/models/__init__.py b/src/diffusers/models/__init__.py index 3f4e49015b..d3bcb3bcee 100755 --- a/src/diffusers/models/__init__.py +++ b/src/diffusers/models/__init__.py @@ -42,6 +42,7 @@ if is_torch_available(): _import_structure["autoencoders.autoencoder_kl_hunyuanvideo15"] = ["AutoencoderKLHunyuanVideo15"] _import_structure["autoencoders.autoencoder_kl_ltx"] = ["AutoencoderKLLTXVideo"] _import_structure["autoencoders.autoencoder_kl_ltx2"] = ["AutoencoderKLLTX2Video"] + _import_structure["autoencoders.autoencoder_kl_ltx2_audio"] = ["AutoencoderKLLTX2Audio"] _import_structure["autoencoders.autoencoder_kl_magvit"] = ["AutoencoderKLMagvit"] _import_structure["autoencoders.autoencoder_kl_mochi"] = ["AutoencoderKLMochi"] _import_structure["autoencoders.autoencoder_kl_qwenimage"] = ["AutoencoderKLQwenImage"] @@ -154,6 +155,7 @@ if TYPE_CHECKING or DIFFUSERS_SLOW_IMPORT: AutoencoderKLHunyuanVideo, AutoencoderKLHunyuanVideo15, AutoencoderKLLTXVideo, + AutoencoderKLLTX2Audio, AutoencoderKLLTX2Video, AutoencoderKLMagvit, AutoencoderKLMochi,