Mirrors/modelscope
1
0
Fork 0
mirror of https://github.com/modelscope/modelscope.git synced 2026-02-24 04:01:10 +01:00
Code Issues Packages Projects Releases Wiki Activity

fix

Browse Source
This commit is contained in:
tastelikefeet
2025-09-28 16:22:12 +08:00
parent 7a37e99b6b
commit ea47332809
1 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
modelscope/utils/hf_util/patcher.py
View File

@@ -244,11 +244,13 @@ def _patch_pretrained_class(all_imported_modules, wrap=False):
extra_allow_file_pattern = list(
(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'
'chat_template.jinja', r'*.json', r'*.py',
r'*.txt', r'*.model'
] # noqa
elif 'Processor' in module_class.__name__:
extra_allow_file_pattern = [
'chat_template.jinja', r'*.json', r'*.py', r'*.txt'
'chat_template.jinja', r'*.json', r'*.py', r'*.txt',
r'*.model'
]
kwargs['allow_file_pattern'] = extra_allow_file_pattern