TypeError: __init__() got an unexpected keyword argument ‘generator’

I am train
I’m trying to tune a whisper-small.en model on aws sagemaker notebook but I get this error when I start training.
notebook kernel: ml.g5.8xlarge
models: openai/whisper-small.en
training notebook example: Fine-Tune Whisper For Multilingual ASR with 🤗 Transformers

can anyone help me to solve this error. Thanks!