diff --git a/modelscope/utils/hf_util/patcher.py b/modelscope/utils/hf_util/patcher.py index f9f9ead3..c6c0527b 100644 --- a/modelscope/utils/hf_util/patcher.py +++ b/modelscope/utils/hf_util/patcher.py @@ -245,12 +245,12 @@ def _patch_pretrained_class(all_imported_modules, wrap=False): (cls.vocab_files_names.values()) if cls is not None and hasattr(cls, 'vocab_files_names') else []) + [ 'chat_template.jinja', r'*.json', r'*.py', - r'*.txt', r'*.model' + r'*.txt', r'*.model', r'*.tiktoken' ] # noqa elif 'Processor' in module_class.__name__: extra_allow_file_pattern = [ 'chat_template.jinja', r'*.json', r'*.py', r'*.txt', - r'*.model' + r'*.model', r'*.tiktoken' ] kwargs['allow_file_pattern'] = extra_allow_file_pattern