bug for ip adapter v2

[ZivKidd]

RuntimeError: Error(s) in loading state_dict for ImageProjModel:
size mismatch for proj.weight: copying a param with shape torch.Size([65536, 768]) from checkpoint, the shape in current model is torch.Size([16384, 768]).
size mismatch for proj.bias: copying a param with shape torch.Size([65536]) from checkpoint, the shape in current model is torch.Size([16384]).

src/flux/xflux_pipeline.py, line 77:
self.improj = ImageProjModel(4096, 768, 4)

looks like the shape is not compatible with v2 model
https://huggingface.co/XLabs-AI/flux-ip-adapter-v2

[mayuelala]

I also met same Problem

[TnoobT]

self.improj = ImageProjModel(cross_attention_dim=4096,clip_embeddings_dim=768,clip_extra_context_tokens=16)