forward() missing 1 required positional argument: 'attention_mask'
napawit44 opened this issue · 6 comments
napawit44 commented
forward() missing 1 required positional argument: 'attention_mask'
xrsrke commented
@napawit44
hi. thanks. i will fix it!!
xrsrke commented
@napawit44 hey. the forward()
of which class? you meant RLHFTrainer.compute_loss()
?
napawit44 commented
napawit44 commented
I ran it on Colab
xrsrke commented
I ran it on Colab
@napawit44 Alright, got it. I'll let you know after pushing the fix
xrsrke commented
@napawit44 hey. i just fixed the bug!!