fix some transformers version missing the chat template definition (#1368)

This commit is contained in:
tastelikefeet
2025-06-11 20:16:24 +08:00
committed by GitHub
parent c82c2f2b0f
commit 86dd6c3434

View File

@@ -243,13 +243,12 @@ def _patch_pretrained_class(all_imported_modules, wrap=False):
from transformers.tokenization_utils import SPECIAL_TOKENS_MAP_FILE from transformers.tokenization_utils import SPECIAL_TOKENS_MAP_FILE
from transformers.tokenization_utils import TOKENIZER_CONFIG_FILE from transformers.tokenization_utils import TOKENIZER_CONFIG_FILE
from transformers.tokenization_utils_base import FULL_TOKENIZER_FILE from transformers.tokenization_utils_base import FULL_TOKENIZER_FILE
from transformers.tokenization_utils_base import CHAT_TEMPLATE_FILE
extra_allow_file_pattern = list( extra_allow_file_pattern = list(
(cls.vocab_files_names.values()) if cls is not None (cls.vocab_files_names.values()) if cls is not None
and hasattr(cls, 'vocab_files_names') else []) + [ and hasattr(cls, 'vocab_files_names') else []) + [
ADDED_TOKENS_FILE, SPECIAL_TOKENS_MAP_FILE, ADDED_TOKENS_FILE, SPECIAL_TOKENS_MAP_FILE,
TOKENIZER_CONFIG_FILE, FULL_TOKENIZER_FILE, TOKENIZER_CONFIG_FILE, FULL_TOKENIZER_FILE,
CHAT_TEMPLATE_FILE, r'*.py' 'chat_template.jinja', r'*.py'
] # noqa ] # noqa
elif 'Processor' in module_class.__name__: elif 'Processor' in module_class.__name__:
from transformers.utils import FEATURE_EXTRACTOR_NAME from transformers.utils import FEATURE_EXTRACTOR_NAME