Error in Attention mask of florence

#12
by Vishwanathan00786 - opened

Attention mask should be of size (24, 1, 582, 582), but is torch.Size([24, 1, 5, 5])

modeling_florence2.py", line 861, in forward
raise ValueError(
ValueError: Attention mask should be of size (25, 1, 10, 10), but is torch.Size([25, 1, 5, 5])

use transformers-4.45.0

Sign up or log in to comment