Mirrors/modelscope
1
0
Fork 0
mirror of https://github.com/modelscope/modelscope.git synced 2025-12-16 16:27:45 +01:00
Code Issues Packages Projects Releases Wiki Activity

lint

Browse Source
This commit is contained in:
yuze.zyz
2024-12-29 13:32:23 +08:00
parent 9f7484bb82
commit 225d1058e3
2 changed files with 5 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
modelscope/hub/api.py
View File

@@ -487,9 +487,7 @@ class HubApi:
raise_for_http_status(r)
return None
def _check_cookie(self,
use_cookies: Union[bool,
CookieJar] = False) -> CookieJar:
def _check_cookie(self, use_cookies: Union[bool, CookieJar] = False) -> CookieJar: # noqa
cookies = None
if isinstance(use_cookies, CookieJar):
cookies = use_cookies

9
modelscope/utils/hf_util.py
View File

@@ -238,12 +238,12 @@ def _patch_pretrained_class():
model_dir = get_model_dir(model_id,
kwargs.pop('ignore_file_pattern', None),
**kwargs)
return kwargs.pop('ori_func')(cls, model, model_dir, *model_args, **kwargs)
return kwargs.pop('ori_func')(cls, model, model_dir, *model_args,
**kwargs)
if PeftModelHF is not None:
PeftModelHF.from_pretrained = partial(
patch_peft_model_id,
ori_func=PeftModelHF.from_pretrained)
patch_peft_model_id, ori_func=PeftModelHF.from_pretrained)
PeftModelForCausalLMHF.from_pretrained = partial(
patch_peft_model_id,
ori_func=PeftModelForCausalLMHF.from_pretrained)
@@ -251,8 +251,7 @@ def _patch_pretrained_class():
patch_peft_model_id,
ori_func=PeftModelForSequenceClassificationHF.from_pretrained)
PeftMixedModelHF.from_pretrained = partial(
patch_peft_model_id,
ori_func=PeftMixedModelHF.from_pretrained)
patch_peft_model_id, ori_func=PeftMixedModelHF.from_pretrained)
def _get_peft_type(cls, model_id, **kwargs):
model_dir = get_model_dir(model_id, ignore_file_pattern, **kwargs)