xrsrke/instructGOOSE

forward() missing 1 required positional argument: 'attention_mask'

napawit44 opened this issue · 6 comments

forward() missing 1 required positional argument: 'attention_mask'

@napawit44
hi. thanks. i will fix it!!

@napawit44 hey. the forward() of which class? you meant RLHFTrainer.compute_loss()?

image

Hi, It is this error from: rewards = reward_model(texts)

I ran it on Colab

I ran it on Colab

@napawit44 Alright, got it. I'll let you know after pushing the fix

@napawit44 hey. i just fixed the bug!!