From 231d8167809e1466c4be57fffc286f7aef584227 Mon Sep 17 00:00:00 2001
From: tastelikefeet <58414341+tastelikefeet@users.noreply.github.com>
Date: Tue, 18 Mar 2025 16:11:42 +0800
Subject: [PATCH] Fix downloading with specified class name (#1270)

---
 modelscope/utils/hf_util/patcher.py | 5 ++---
 1 file changed, 2 insertions(+), 3 deletions(-)

diff --git a/modelscope/utils/hf_util/patcher.py b/modelscope/utils/hf_util/patcher.py
index ecc33662..02afb758 100644
--- a/modelscope/utils/hf_util/patcher.py
+++ b/modelscope/utils/hf_util/patcher.py
@@ -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 = {