export/expose SD1Autoencoder and SDXLAutoencoder + some formatting

finegrain-ai · Sep 10, 2024 · 4eceda8 · 4eceda8
1 parent 602c63c
commit 4eceda8
Showing 1 changed file with 6 additions and 6 deletions.
diff --git 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_xl import (
     ControlLoraAdapter,
     DoubleTextEncoder,
+    SDXLAutoencoder,
     SDXLIPAdapter,
     SDXLLcmAdapter,
     SDXLT2IAdapter,
@@ -30,11 +28,13 @@
     "StableDiffusion_1",
     "StableDiffusion_1_Inpainting",
     "SD1UNet",
+    "SD1Autoencoder",
     "SD1ControlnetAdapter",
     "SD1IPAdapter",
     "SD1T2IAdapter",
     "SD1ELLAAdapter",
     "SDXLUNet",
+    "SDXLAutoencoder",
     "DoubleTextEncoder",
     "SDXLIPAdapter",
     "SDXLLcmAdapter",