From a6a9c8b972adc7749ab95127dcd81f35d5ddf280 Mon Sep 17 00:00:00 2001 From: hugojarkoff Date: Wed, 17 Jan 2024 15:53:50 +0100 Subject: [PATCH] Fix Value dimension in ImageCrossAttention --- 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 5298556..8d6378b 100644 --- a/src/refiners/foundationals/latent_diffusion/image_prompt.py +++ b/src/refiners/foundationals/latent_diffusion/image_prompt.py @@ -252,7 +252,7 @@ class ImageCrossAttention(fl.Chain): fl.Chain( fl.UseContext(context="ip_adapter", key="clip_image_embedding"), fl.Linear( - in_features=text_cross_attention.key_embedding_dim, + in_features=text_cross_attention.value_embedding_dim, out_features=text_cross_attention.inner_dim, bias=text_cross_attention.use_bias, device=text_cross_attention.device,