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 lint

Browse Source
This commit is contained in:
suluyan
2025-03-21 11:36:24 +08:00
parent 8292fd9262
commit c47d0135ea
1 changed files with 13 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -25,9 +25,19 @@ def get_all_imported_modules():
"""Find all modules in transformers/peft/diffusers"""
all_imported_modules = []
transformers_include_names = [
'Auto.*', 'T5.*', 'BitsAndBytesConfig', 'GenerationConfig', 'Awq.*',
'GPTQ.*', 'BatchFeature', 'Qwen.*', 'Llama.*', 'PretrainedConfig',
'PreTrainedTokenizer', 'PreTrainedModel', 'PreTrainedTokenizerFast',
'Auto.*',
'T5.*',
'BitsAndBytesConfig',
'GenerationConfig',
'Awq.*',
'GPTQ.*',
'BatchFeature',
'Qwen.*',
'Llama.*',
'PretrainedConfig',
'PreTrainedTokenizer',
'PreTrainedModel',
'PreTrainedTokenizerFast',
]
peft_include_names = ['.*PeftModel.*', '.*Config']
diffusers_include_names = [