diff --git a/pyproject.toml b/pyproject.toml index ed41d8f67f..1c3b82a699 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -186,6 +186,8 @@ filterwarnings = [ '''ignore:.*`load_state_dict` is deprecated and *:FutureWarning''', # Ignore NeptuneLogger warnings '''ignore:.*NVML Shared Library Not Found. GPU usage metrics may not be reported.*''', + # Ignore HuggingFace future warnings + '''ignore:.*`clean_up_tokenization_spaces` was not set.*:FutureWarning''', ] # Coverage diff --git a/setup.py b/setup.py index a6c19eb21e..b702ca79db 100644 --- a/setup.py +++ b/setup.py @@ -179,7 +179,7 @@ def package_files(prefix: str, directory: str, extension: str): ] extra_deps['nlp'] = [ - 'transformers>=4.11,!=4.34.0,<4.44', + 'transformers>=4.11,!=4.34.0,<4.45', 'datasets>=2.4,<3', 'huggingface-hub>=0.21.2,<0.25', ]