From 402d3105b4b486ca884492f1e9985dbb60622fcb Mon Sep 17 00:00:00 2001 From: Pierre Chapuis Date: Fri, 9 Feb 2024 16:29:54 +0100 Subject: [PATCH] support multiple IP adapter inputs as tensor --- src/refiners/foundationals/latent_diffusion/image_prompt.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/refiners/foundationals/latent_diffusion/image_prompt.py b/src/refiners/foundationals/latent_diffusion/image_prompt.py index e90337c..75fc5a0 100644 --- a/src/refiners/foundationals/latent_diffusion/image_prompt.py +++ b/src/refiners/foundationals/latent_diffusion/image_prompt.py @@ -440,7 +440,7 @@ class IPAdapter(Generic[T], fl.Chain, Adapter[T]): self.set_context("ip_adapter", {"clip_image_embedding": image_embedding}) @overload - def compute_clip_image_embedding(self, image_prompt: Tensor) -> Tensor: + def compute_clip_image_embedding(self, image_prompt: Tensor, weights: list[float] | None = None) -> Tensor: ... @overload