diff --git a/docs/reference/foundationals/latent_diffusion.md b/docs/reference/foundationals/latent_diffusion.md index 85ae5d6..b735508 100644 --- a/docs/reference/foundationals/latent_diffusion.md +++ b/docs/reference/foundationals/latent_diffusion.md @@ -9,3 +9,5 @@ ::: refiners.foundationals.latent_diffusion.lora ::: refiners.foundationals.latent_diffusion.image_prompt + +::: refiners.foundationals.latent_diffusion.style_aligned diff --git a/src/refiners/foundationals/latent_diffusion/__init__.py b/src/refiners/foundationals/latent_diffusion/__init__.py index 35e1b9a..dfba3dd 100644 --- a/src/refiners/foundationals/latent_diffusion/__init__.py +++ b/src/refiners/foundationals/latent_diffusion/__init__.py @@ -21,6 +21,7 @@ from refiners.foundationals.latent_diffusion.stable_diffusion_xl import ( SDXLUNet, StableDiffusion_XL, ) +from refiners.foundationals.latent_diffusion.style_aligned import StyleAlignedAdapter __all__ = [ "StableDiffusion_1", @@ -39,4 +40,5 @@ __all__ = [ "LatentDiffusionAutoencoder", "SDFreeUAdapter", "StableDiffusion_XL", + "StyleAlignedAdapter", ]