Closed mranzinger closed 1 month ago
Yep opening a PR asap cc @xenova !
I think this is still broken. self.position_embedding
is of type nn.Embedding
, which means it doesn't have a .shape
variable.
I think you want to bring back
position_embeddings = self.position_embedding.weight.unsqueeze(0)
Fixed properly this time sorry!
Originally posted by @mranzinger in c6d2848