tiny-smart / in-context-matting

41 stars 1 forks source link

An error occurred when evaluating(ValueError: Incompatible shapes for attention inputs) #5

Open BRONYA-818 opened 5 months ago

BRONYA-818 commented 5 months ago

An error occurred when evaluating: ValueError: Incompatible shapes for attention inputs: query.shape: torch.Size([20, 9216, 64]) key.shape : torch.Size([40, 77, 64]) value.shape: torch.Size([40, 77, 64]) HINT: We don't support broadcasting, please use expand yourself before calling memory_efficient_attention if you need to what should I do to solve this?

guohe369 commented 3 weeks ago

It seems that the batch sizes of images and prompts are not match. Could you provide more details?