mirror of
https://github.com/finegrain-ai/refiners.git
synced 2024-11-09 23:12:02 +00:00
Fix Value dimension in ImageCrossAttention
This commit is contained in:
parent
7839c54ae8
commit
a6a9c8b972
|
@ -252,7 +252,7 @@ class ImageCrossAttention(fl.Chain):
|
||||||
fl.Chain(
|
fl.Chain(
|
||||||
fl.UseContext(context="ip_adapter", key="clip_image_embedding"),
|
fl.UseContext(context="ip_adapter", key="clip_image_embedding"),
|
||||||
fl.Linear(
|
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,
|
out_features=text_cross_attention.inner_dim,
|
||||||
bias=text_cross_attention.use_bias,
|
bias=text_cross_attention.use_bias,
|
||||||
device=text_cross_attention.device,
|
device=text_cross_attention.device,
|
||||||
|
|
Loading…
Reference in a new issue