Hugging Face's logo

MU-NLPC
/
whisper-small-audio-captioning

Transformers
PyTorch
English
whisper
Eval Results (legacy)
Model card Files
xet
 Community
7

Update model_class.py

#5
by geekweilai - opened
base: refs/heads/main
←
from: refs/pr/5
Discussion Files changed
+2
-0
geekweilai
•
edited Jan 27, 2024

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.
Update model_class.pyc0a6c8d4
prompteus
 NLP Centre, Faculty of Informatics, Masaryk University org

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

Sign up or log in to comment