From 3196a26ed614c156074539ad422ae1b5e72ef35c Mon Sep 17 00:00:00 2001
From: Laurent <laurent@lagon.tech>
Date: Sun, 8 Sep 2024 13:48:47 +0000
Subject: [PATCH] export/expose SD1Autoencoder and SDXLAutoencoder + some
 formatting

---
 .../foundationals/latent_diffusion/__init__.py       | 12 ++++++------
 1 file changed, 6 insertions(+), 6 deletions(-)

diff --git a/src/refiners/foundationals/latent_diffusion/__init__.py b/src/refiners/foundationals/latent_diffusion/__init__.py
index f7290e2..2df67b1 100644
--- a/src/refiners/foundationals/latent_diffusion/__init__.py
+++ b/src/refiners/foundationals/latent_diffusion/__init__.py
@@ -1,12 +1,9 @@
-from refiners.foundationals.clip.text_encoder import (
-    CLIPTextEncoderL,
-)
-from refiners.foundationals.latent_diffusion.auto_encoder import (
-    LatentDiffusionAutoencoder,
-)
+from refiners.foundationals.clip.text_encoder import CLIPTextEncoderL
+from refiners.foundationals.latent_diffusion.auto_encoder import LatentDiffusionAutoencoder
 from refiners.foundationals.latent_diffusion.freeu import SDFreeUAdapter
 from refiners.foundationals.latent_diffusion.solvers import DPMSolver, LCMSolver, Solver
 from refiners.foundationals.latent_diffusion.stable_diffusion_1 import (
+    SD1Autoencoder,
     SD1ControlnetAdapter,
     SD1ELLAAdapter,
     SD1IPAdapter,
@@ -18,6 +15,7 @@ from refiners.foundationals.latent_diffusion.stable_diffusion_1 import (
 from refiners.foundationals.latent_diffusion.stable_diffusion_xl import (
     ControlLoraAdapter,
     DoubleTextEncoder,
+    SDXLAutoencoder,
     SDXLIPAdapter,
     SDXLLcmAdapter,
     SDXLT2IAdapter,
@@ -30,11 +28,13 @@ __all__ = [
     "StableDiffusion_1",
     "StableDiffusion_1_Inpainting",
     "SD1UNet",
+    "SD1Autoencoder",
     "SD1ControlnetAdapter",
     "SD1IPAdapter",
     "SD1T2IAdapter",
     "SD1ELLAAdapter",
     "SDXLUNet",
+    "SDXLAutoencoder",
     "DoubleTextEncoder",
     "SDXLIPAdapter",
     "SDXLLcmAdapter",