For anyone running into the same issue- I was on an older version (4.23.1
) where forward()
indeed did not have attention_mask
as an input argument. Upgrading to the latest version fixed it!
For anyone running into the same issue- I was on an older version (4.23.1
) where forward()
indeed did not have attention_mask
as an input argument. Upgrading to the latest version fixed it!