TypeError: forward() got an unexpected keyword argument 'attention_mask'

#18
by jkb0722 - opened

Hi,
After I use the tokenizer on my data, I pass the attention mask to the DNABERT2 model since my sequences aren't of the same length. However, when I use the keyword attention_mask, I receive an error saying that the keyword is unexpected. However, it appears to be an optional argument when I try to look into the code. Could you please explain how I can properly pass the attention_mask?

Thank you!

Sign up or log in to comment