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

import ControlLoraAdapter part of latent_diffusion

Browse source
This commit is contained in:
Cédric Deltheil 2024-02-19 13:05:36 +00:00 committed by Cédric Deltheil
parent 8b799bf3ad
commit 5ab5d7fd1c
2 changed files with 3 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -15,6 +15,7 @@ from refiners.foundationals.latent_diffusion.stable_diffusion_1 import (
StableDiffusion_1_Inpainting, StableDiffusion_1_Inpainting,
) )
from refiners.foundationals.latent_diffusion.stable_diffusion_xl import ( from refiners.foundationals.latent_diffusion.stable_diffusion_xl import (
ControlLoraAdapter,
DoubleTextEncoder, DoubleTextEncoder,
SDXLIPAdapter, SDXLIPAdapter,
SDXLT2IAdapter, SDXLT2IAdapter,
@ -41,4 +42,5 @@ __all__ = [
"SDFreeUAdapter", "SDFreeUAdapter",
"StableDiffusion_XL", "StableDiffusion_XL",
"StyleAlignedAdapter", "StyleAlignedAdapter",
"ControlLoraAdapter",
] ]

2
tests/e2e/test_diffusion.py
View file

@ -12,6 +12,7 @@ from refiners.fluxion.layers.attentions import ScaledDotProductAttention
from refiners.fluxion.utils import image_to_tensor, load_from_safetensors, load_tensors, manual_seed, no_grad from refiners.fluxion.utils import image_to_tensor, load_from_safetensors, load_tensors, manual_seed, no_grad
from refiners.foundationals.clip.concepts import ConceptExtender from refiners.foundationals.clip.concepts import ConceptExtender
from refiners.foundationals.latent_diffusion import ( from refiners.foundationals.latent_diffusion import (
ControlLoraAdapter,
SD1ControlnetAdapter, SD1ControlnetAdapter,
SD1IPAdapter, SD1IPAdapter,
SD1T2IAdapter, SD1T2IAdapter,
@ -28,7 +29,6 @@ from refiners.foundationals.latent_diffusion.reference_only_control import Refer
from refiners.foundationals.latent_diffusion.restart import Restart from refiners.foundationals.latent_diffusion.restart import Restart
from refiners.foundationals.latent_diffusion.solvers import DDIM, Euler, NoiseSchedule from refiners.foundationals.latent_diffusion.solvers import DDIM, Euler, NoiseSchedule
from refiners.foundationals.latent_diffusion.stable_diffusion_1.multi_diffusion import SD1MultiDiffusion from refiners.foundationals.latent_diffusion.stable_diffusion_1.multi_diffusion import SD1MultiDiffusion
from refiners.foundationals.latent_diffusion.stable_diffusion_xl.control_lora import ControlLoraAdapter
from refiners.foundationals.latent_diffusion.stable_diffusion_xl.model import StableDiffusion_XL from refiners.foundationals.latent_diffusion.stable_diffusion_xl.model import StableDiffusion_XL
from refiners.foundationals.latent_diffusion.style_aligned import StyleAlignedAdapter from refiners.foundationals.latent_diffusion.style_aligned import StyleAlignedAdapter
from tests.utils import ensure_similar_images from tests.utils import ensure_similar_images