Finegrain/refiners
1
0
Fork 0
mirror of https://github.com/finegrain-ai/refiners.git synced 2024-09-19 19:05:28 +00:00
Code Issues Actions 9 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,
)
from refiners.foundationals.latent_diffusion.stable_diffusion_xl import (
ControlLoraAdapter,
DoubleTextEncoder,
SDXLIPAdapter,
SDXLT2IAdapter,
@ -41,4 +42,5 @@ __all__ = [
"SDFreeUAdapter",
"StableDiffusion_XL",
"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.foundationals.clip.concepts import ConceptExtender
from refiners.foundationals.latent_diffusion import (
ControlLoraAdapter,
SD1ControlnetAdapter,
SD1IPAdapter,
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.solvers import DDIM, Euler, NoiseSchedule
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.style_aligned import StyleAlignedAdapter
from tests.utils import ensure_similar_images