diff --git a/src/refiners/foundationals/latent_diffusion/stable_diffusion_xl/__init__.py b/src/refiners/foundationals/latent_diffusion/stable_diffusion_xl/__init__.py
index 43f2c36..d2678c0 100644
--- a/src/refiners/foundationals/latent_diffusion/stable_diffusion_xl/__init__.py
+++ b/src/refiners/foundationals/latent_diffusion/stable_diffusion_xl/__init__.py
@@ -1,13 +1,14 @@
 from refiners.foundationals.latent_diffusion.stable_diffusion_xl.image_prompt import SDXLIPAdapter
-from refiners.foundationals.latent_diffusion.stable_diffusion_xl.model import StableDiffusion_XL
+from refiners.foundationals.latent_diffusion.stable_diffusion_xl.model import SDXLAutoencoder, StableDiffusion_XL
 from refiners.foundationals.latent_diffusion.stable_diffusion_xl.t2i_adapter import SDXLT2IAdapter
 from refiners.foundationals.latent_diffusion.stable_diffusion_xl.text_encoder import DoubleTextEncoder
 from refiners.foundationals.latent_diffusion.stable_diffusion_xl.unet import SDXLUNet
 
 __all__ = [
     "SDXLUNet",
-    "DoubleTextEncoder",
     "StableDiffusion_XL",
+    "DoubleTextEncoder",
+    "SDXLAutoencoder",
     "SDXLIPAdapter",
     "SDXLT2IAdapter",
 ]