From 8bab4b9bc16eeffeea576ba96e31ab35805a4670 Mon Sep 17 00:00:00 2001
From: tastelikefeet <58414341+tastelikefeet@users.noreply.github.com>
Date: Sun, 31 May 2026 16:11:21 +0800
Subject: [PATCH] Fix bug: 1. patch_context cannot revert classmethod 2.
 kernels testcase miss trust_remote_code and revision (#1729)

---
 modelscope/utils/hf_util/patcher.py | 34 ++++++++++++++++-------------
 tests/utils/test_hf_util_kernels.py |  6 +++--
 2 files changed, 23 insertions(+), 17 deletions(-)

diff --git a/modelscope/utils/hf_util/patcher.py b/modelscope/utils/hf_util/patcher.py
index 59fefee0..e58e84b6 100644
--- a/modelscope/utils/hf_util/patcher.py
+++ b/modelscope/utils/hf_util/patcher.py
@@ -478,6 +478,20 @@ def _patch_pretrained_class(all_imported_modules, wrap=False):
 
 
 def _unpatch_pretrained_class(all_imported_modules):
+    # The patcher captured `var.from_pretrained` via descriptor access, which
+    # returns a bound method. Re-wrap as classmethod so subclasses still get
+    # `cls` bound to themselves (not to the ancestor that was patched).
+    def _restore(var, attr, origin_attr):
+        origin = getattr(var, origin_attr)
+        if isinstance(origin, MethodType):
+            setattr(var, attr, classmethod(origin.__func__))
+        else:
+            setattr(var, attr, origin)
+        try:
+            delattr(var, origin_attr)
+        except Exception:  # noqa
+            pass
+
     for var in all_imported_modules:
         if var is None:
             continue
@@ -489,23 +503,11 @@ def _unpatch_pretrained_class(all_imported_modules):
         except:  # noqa
             continue
         if has_from_pretrained and hasattr(var, '_from_pretrained_origin'):
-            var.from_pretrained = var._from_pretrained_origin
-            try:
-                delattr(var, '_from_pretrained_origin')
-            except:  # noqa
-                pass
+            _restore(var, 'from_pretrained', '_from_pretrained_origin')
         if has_get_peft_type and hasattr(var, '_get_peft_type_origin'):
-            var._get_peft_type = var._get_peft_type_origin
-            try:
-                delattr(var, '_get_peft_type_origin')
-            except:  # noqa
-                pass
+            _restore(var, '_get_peft_type', '_get_peft_type_origin')
         if has_get_config_dict and hasattr(var, '_get_config_dict_origin'):
-            var.get_config_dict = var._get_config_dict_origin
-            try:
-                delattr(var, '_get_config_dict_origin')
-            except:  # noqa
-                pass
+            _restore(var, 'get_config_dict', '_get_config_dict_origin')
 
     from transformers import dynamic_module_utils
     if hasattr(dynamic_module_utils, 'origin_get_class_from_dynamic_module'):
@@ -565,6 +567,8 @@ class _MsKernelApi:
                           local_files_only=False,
                           **kwargs):
         from modelscope import snapshot_download as ms_snapshot_download
+        if kwargs.get('repo_type') == 'kernel':
+            kwargs['repo_type'] = 'model'
         return ms_snapshot_download(
             repo_id,
             revision=_ms_revision(revision),
diff --git a/tests/utils/test_hf_util_kernels.py b/tests/utils/test_hf_util_kernels.py
index 3cafb17f..fb537ae6 100644
--- a/tests/utils/test_hf_util_kernels.py
+++ b/tests/utils/test_hf_util_kernels.py
@@ -256,7 +256,8 @@ class TinyGradRMSIntegrationTest(_KernelsTestBase):
         import modelscope
         # Routes through `try_import_from_hf` and scopes the ModelScope
         # monkey-patch to this single call.
-        module = modelscope.get_kernel(self.REPO)
+        module = modelscope.get_kernel(
+            self.REPO, trust_remote_code=True, revision='master')
         self.assertIsNotNone(module)
         # Wrapper must leave `_get_hf_api` restored afterwards.
         self.assertIs(self.kernels_utils._get_hf_api, self.original_get_hf_api)
@@ -264,7 +265,8 @@ class TinyGradRMSIntegrationTest(_KernelsTestBase):
     def test_patch_hub_then_kernels_get_kernel(self):
         with _isolate_hub_patches(), patch_context():
             from kernels import get_kernel
-            module = get_kernel(self.REPO)
+            module = get_kernel(
+                self.REPO, trust_remote_code=True, revision='master')
             self.assertIsNotNone(module)
         # `patch_context` rolled the kernels patch back on exit.
         self.assertIs(self.kernels_utils._get_hf_api, self.original_get_hf_api)