From f2bcb7f45e4e140281104210ed2bed42033764ed Mon Sep 17 00:00:00 2001 From: Laurent Date: Fri, 2 Feb 2024 11:12:33 +0000 Subject: [PATCH] (mkdocstrings) export `SDXLAutoencoder` in `src/refiners/foundationals/latent_diffusion/stable_diffusion_xl/__init__.py` --- .../latent_diffusion/stable_diffusion_xl/__init__.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) 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", ]