Unverified Commit 50540e18 authored by Yih-Dar's avatar Yih-Dar Committed by GitHub
Browse files

Update `huggingface_hub` version (#21212)



* update huggingface_hub version

* revert changes in setup.py
Co-authored-by: default avatarydshieh <ydshieh@users.noreply.github.com>
parent 202d6863
......@@ -156,10 +156,10 @@ class Wav2Vec2ProcessorWithLM(ProcessorMixin):
# make sure that only relevant filenames are downloaded
language_model_filenames = os.path.join(BeamSearchDecoderCTC._LANGUAGE_MODEL_SERIALIZED_DIRECTORY, "*")
alphabet_filename = BeamSearchDecoderCTC._ALPHABET_SERIALIZED_FILENAME
allow_regex = [language_model_filenames, alphabet_filename]
allow_patterns = [language_model_filenames, alphabet_filename]
decoder = BeamSearchDecoderCTC.load_from_hf_hub(
pretrained_model_name_or_path, allow_regex=allow_regex, **kwargs
pretrained_model_name_or_path, allow_patterns=allow_patterns, **kwargs
)
# set language model attributes
......
......@@ -849,8 +849,8 @@ def pipeline(
BeamSearchDecoderCTC._LANGUAGE_MODEL_SERIALIZED_DIRECTORY, "*"
)
alphabet_filename = BeamSearchDecoderCTC._ALPHABET_SERIALIZED_FILENAME
allow_regex = [language_model_glob, alphabet_filename]
decoder = BeamSearchDecoderCTC.load_from_hf_hub(model_name, allow_regex=allow_regex)
allow_patterns = [language_model_glob, alphabet_filename]
decoder = BeamSearchDecoderCTC.load_from_hf_hub(model_name, allow_patterns=allow_patterns)
kwargs["decoder"] = decoder
except ImportError as e:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment