From 5ab5d7fd1c6f30a8d5336df827d7c27c2618d160 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?C=C3=A9dric=20Deltheil?= Date: Mon, 19 Feb 2024 13:05:36 +0000 Subject: [PATCH] import ControlLoraAdapter part of latent_diffusion --- src/refiners/foundationals/latent_diffusion/__init__.py | 2 ++ tests/e2e/test_diffusion.py | 2 +- 2 files changed, 3 insertions(+), 1 deletion(-) diff --git a/src/refiners/foundationals/latent_diffusion/__init__.py b/src/refiners/foundationals/latent_diffusion/__init__.py index dfba3dd..208a4fb 100644 --- a/src/refiners/foundationals/latent_diffusion/__init__.py +++ b/src/refiners/foundationals/latent_diffusion/__init__.py @@ -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", ] diff --git a/tests/e2e/test_diffusion.py b/tests/e2e/test_diffusion.py index 418cfb9..9631d83 100644 --- a/tests/e2e/test_diffusion.py +++ b/tests/e2e/test_diffusion.py @@ -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