Finegrain/refiners
1
0
Fork 0
mirror of https://github.com/finegrain-ai/refiners.git synced 2024-11-21 21:58:47 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

export/expose SD1Autoencoder and SDXLAutoencoder + some formatting

Browse source
This commit is contained in:
Laurent 2024-09-08 13:48:47 +00:00
parent 04fd98ce20
commit 3196a26ed6
No known key found for this signature in database
1 changed files with 6 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
src/refiners/foundationals/latent_diffusion/__init__.py
View file

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