Update model_class.py
#5
by
geekweilai
- opened
Hi, there.
I meet an error that the WhisperForAudioCaptioning.forward() method is receiving an unexpected keyword argument decoder_position_ids.
This argument is not defined in the forward() method of the WhisperForAudioCaptioning class. To fix this issue, I modify the forward() method of the WhisperForAudioCaptioning class to accept decoder_position_ids as an argument.
Waiting for your reply.
Hi,
Thank you for your contribution.
The decoder_position_ids
parameter didn't exist in transformers.WhisperForConditionalGeneration
at the time we wrote this code.
The change seems good to me, I'm merging it. Thanks.
prompteus
changed pull request status to
merged