Missmatch tensor size with vit5-large

vietai / ViT5

MIT License

59 stars 9 forks source link

Missmatch tensor size with vit5-large #3

Closed batman-do closed 2 years ago

batman-do commented 2 years ago

How to fix this error beacause i train with mT5 no appear error this ? Thanks guys !

heraclex12 commented 2 years ago

Hi @Dean98AI ,

Thank you for letting us know about this issue. There was something wrong with our config, but it was fixed now. You can try to load our large model again.