Skip to content
2 changes: 2 additions & 0 deletions src/diffusers/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -157,6 +157,7 @@
"AutoencoderKLWan",
"AutoencoderOobleck",
"AutoencoderTiny",
"AutoencoderVidTok",
"AutoModel",
"CacheMixin",
"ChromaTransformer2DModel",
Expand Down Expand Up @@ -770,6 +771,7 @@
AutoencoderKLWan,
AutoencoderOobleck,
AutoencoderTiny,
AutoencoderVidTok,
AutoModel,
CacheMixin,
ChromaTransformer2DModel,
Expand Down
2 changes: 2 additions & 0 deletions src/diffusers/models/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -41,6 +41,7 @@
_import_structure["autoencoders.autoencoder_kl_wan"] = ["AutoencoderKLWan"]
_import_structure["autoencoders.autoencoder_oobleck"] = ["AutoencoderOobleck"]
_import_structure["autoencoders.autoencoder_tiny"] = ["AutoencoderTiny"]
_import_structure["autoencoders.autoencoder_vidtok"] = ["AutoencoderVidTok"]
_import_structure["autoencoders.consistency_decoder_vae"] = ["ConsistencyDecoderVAE"]
_import_structure["autoencoders.vq_model"] = ["VQModel"]
_import_structure["cache_utils"] = ["CacheMixin"]
Expand Down Expand Up @@ -127,6 +128,7 @@
AutoencoderKLWan,
AutoencoderOobleck,
AutoencoderTiny,
AutoencoderVidTok,
ConsistencyDecoderVAE,
VQModel,
)
Expand Down
Loading