Fix following https://github.com/huggingface/transformers/commit/2da8853775b61cde0894dee17c6c713aba711688 08b2a7b kokiryu commited on Dec 1, 2023