Function _unpatch_pretrained_class

modelscope/utils/hf_util/patcher.py:480–517

(all_imported_modules)

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

        try:
            has_from_pretrained = hasattr(var, 'from_pretrained')
            has_get_peft_type = hasattr(var, '_get_peft_type')
            has_get_config_dict = hasattr(var, 'get_config_dict')
        except:  # noqa
            continue
        if has_from_pretrained and hasattr(var, '_from_pretrained_origin'):
            _restore(var, 'from_pretrained', '_from_pretrained_origin')
        if has_get_peft_type and hasattr(var, '_get_peft_type_origin'):
            _restore(var, '_get_peft_type', '_get_peft_type_origin')
        if has_get_config_dict and hasattr(var, '_get_config_dict_origin'):
            _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'):
        dynamic_module_utils.get_class_from_dynamic_module = dynamic_module_utils.origin_get_class_from_dynamic_module
        from transformers.models.auto import configuration_auto
        configuration_auto.get_class_from_dynamic_module = dynamic_module_utils.origin_get_class_from_dynamic_module
        delattr(dynamic_module_utils, 'origin_get_class_from_dynamic_module')


def _patch_kernels():