Fix tokenizer warning filtering for processors (#34500)
This commit is contained in:
@@ -48,6 +48,7 @@ from .tokenizer import (
|
|||||||
_TOKENIZERS_BACKEND,
|
_TOKENIZERS_BACKEND,
|
||||||
_fix_added_tokens_encoding,
|
_fix_added_tokens_encoding,
|
||||||
_fix_special_tokens_pattern,
|
_fix_special_tokens_pattern,
|
||||||
|
_install_tokenizer_warnings_filter,
|
||||||
)
|
)
|
||||||
|
|
||||||
_IMAGE_PROCESSOR_BACKENDS = {"auto", "torchvision", "pil"}
|
_IMAGE_PROCESSOR_BACKENDS = {"auto", "torchvision", "pil"}
|
||||||
@@ -368,6 +369,8 @@ def get_processor(
|
|||||||
else:
|
else:
|
||||||
processor.tokenizer = tokenizer
|
processor.tokenizer = tokenizer
|
||||||
|
|
||||||
|
_install_tokenizer_warnings_filter(tokenizer)
|
||||||
|
|
||||||
if tokenizer.chat_template is None:
|
if tokenizer.chat_template is None:
|
||||||
local_path = download_from_hf(
|
local_path = download_from_hf(
|
||||||
tokenizer_name, allow_patterns=["*.json", "*.jinja", "*.model"]
|
tokenizer_name, allow_patterns=["*.json", "*.jinja", "*.model"]
|
||||||
|
|||||||
@@ -132,6 +132,15 @@ class TokenizerWarningsFilter(logging.Filter):
|
|||||||
return "Calling super().encode with" not in record.getMessage()
|
return "Calling super().encode with" not in record.getMessage()
|
||||||
|
|
||||||
|
|
||||||
|
_tokenizer_warnings_filter = TokenizerWarningsFilter()
|
||||||
|
|
||||||
|
|
||||||
|
def _install_tokenizer_warnings_filter(tokenizer):
|
||||||
|
logging.getLogger(tokenizer.__class__.__module__).addFilter(
|
||||||
|
_tokenizer_warnings_filter
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
# ---------------------------------------------------------------------------
|
# ---------------------------------------------------------------------------
|
||||||
# Helpers for get_tokenizer
|
# Helpers for get_tokenizer
|
||||||
# ---------------------------------------------------------------------------
|
# ---------------------------------------------------------------------------
|
||||||
@@ -169,9 +178,6 @@ def _auto_tokenizer_from_pretrained(tokenizer_name, *args, **common_kwargs):
|
|||||||
tokenizer = AutoTokenizer.from_pretrained(
|
tokenizer = AutoTokenizer.from_pretrained(
|
||||||
tokenizer_name, *args, **common_kwargs
|
tokenizer_name, *args, **common_kwargs
|
||||||
)
|
)
|
||||||
logging.getLogger(tokenizer.__class__.__module__).addFilter(
|
|
||||||
TokenizerWarningsFilter()
|
|
||||||
)
|
|
||||||
return tokenizer
|
return tokenizer
|
||||||
except TypeError as e:
|
except TypeError as e:
|
||||||
err_msg = (
|
err_msg = (
|
||||||
@@ -419,6 +425,7 @@ def _fix_special_tokens_pattern(tokenizer):
|
|||||||
|
|
||||||
def _apply_post_load_fixes(tokenizer, tokenizer_name, revision):
|
def _apply_post_load_fixes(tokenizer, tokenizer_name, revision):
|
||||||
"""Apply all post-load patches and return the final tokenizer."""
|
"""Apply all post-load patches and return the final tokenizer."""
|
||||||
|
_install_tokenizer_warnings_filter(tokenizer)
|
||||||
_fix_v5_tokenizer_components(tokenizer, tokenizer_name, revision)
|
_fix_v5_tokenizer_components(tokenizer, tokenizer_name, revision)
|
||||||
_fix_v5_add_bos_eos_token(tokenizer, tokenizer_name, revision)
|
_fix_v5_add_bos_eos_token(tokenizer, tokenizer_name, revision)
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user