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): ...@@ -156,10 +156,10 @@ class Wav2Vec2ProcessorWithLM(ProcessorMixin):
# make sure that only relevant filenames are downloaded # make sure that only relevant filenames are downloaded
language_model_filenames = os.path.join(BeamSearchDecoderCTC._LANGUAGE_MODEL_SERIALIZED_DIRECTORY, "*") language_model_filenames = os.path.join(BeamSearchDecoderCTC._LANGUAGE_MODEL_SERIALIZED_DIRECTORY, "*")
alphabet_filename = BeamSearchDecoderCTC._ALPHABET_SERIALIZED_FILENAME 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( 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 # set language model attributes
......
...@@ -849,8 +849,8 @@ def pipeline( ...@@ -849,8 +849,8 @@ def pipeline(
BeamSearchDecoderCTC._LANGUAGE_MODEL_SERIALIZED_DIRECTORY, "*" BeamSearchDecoderCTC._LANGUAGE_MODEL_SERIALIZED_DIRECTORY, "*"
) )
alphabet_filename = BeamSearchDecoderCTC._ALPHABET_SERIALIZED_FILENAME alphabet_filename = BeamSearchDecoderCTC._ALPHABET_SERIALIZED_FILENAME
allow_regex = [language_model_glob, alphabet_filename] allow_patterns = [language_model_glob, alphabet_filename]
decoder = BeamSearchDecoderCTC.load_from_hf_hub(model_name, allow_regex=allow_regex) decoder = BeamSearchDecoderCTC.load_from_hf_hub(model_name, allow_patterns=allow_patterns)
kwargs["decoder"] = decoder kwargs["decoder"] = decoder
except ImportError as e: 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