diff --git a/src/transformers/models/whisper/convert_openai_to_hf.py b/src/transformers/models/whisper/convert_openai_to_hf.py index 8d29a8434ac..702fbd0204a 100755 --- a/src/transformers/models/whisper/convert_openai_to_hf.py +++ b/src/transformers/models/whisper/convert_openai_to_hf.py @@ -21,7 +21,7 @@ import os import tempfile import urllib import warnings -from typing import Any, Optional, Tuple +from typing import Any, List, Optional, Tuple import torch from huggingface_hub.utils import insecure_hashlib @@ -252,7 +252,7 @@ def convert_openai_whisper_to_tfms( # Adapted from https://github.com/openai/tiktoken/issues/60#issuecomment-1499977960 -def _bpe(mergeable_ranks, token: bytes, max_rank=None) -> list[bytes]: +def _bpe(mergeable_ranks, token: bytes, max_rank=None) -> List[bytes]: parts = [bytes([b]) for b in token] while True: min_idx = None