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 downloading with specified class name (#1270)

Browse Source
This commit is contained in:
tastelikefeet
2025-03-18 16:11:42 +08:00
committed by GitHub
parent 51655d3526
commit 231d816780
1 changed files with 2 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -321,9 +321,8 @@ def _patch_pretrained_class(all_imported_modules, wrap=False):
if var is None or not hasattr(var, '__name__'):
continue
name = var.__name__
need_model = 'model' in name.lower() or 'processor' in name.lower(
) or 'extractor' in name.lower() or 'pipeline' in name.lower()
if need_model:
skip_model = 'tokenizer' in name.lower() or 'config' in name.lower()
if not skip_model:
ignore_file_pattern_kwargs = {}
else:
ignore_file_pattern_kwargs = {