Mirrors/modelscope
1
0
Fork 0
mirror of https://github.com/modelscope/modelscope.git synced 2026-09-01 19:49:03 +02:00
Code Issues Packages Projects Releases Wiki Activity

add more diffusers imports (#1265)

Browse Source
This commit is contained in:
tastelikefeet
2025-03-15 13:39:24 +08:00
committed by GitHub
parent 3274bb1e63
commit 51655d3526
1 changed files with 5 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -31,7 +31,11 @@ def get_all_imported_modules():
'Pipeline'
]
peft_include_names = ['.*PeftModel.*', '.*Config']
diffusers_include_names = ['^(?!TF|Flax).*Pipeline$']
diffusers_include_names = [
'^(?!TF|Flax).*Pipeline$', '^(?!TF|Flax).*Autoencoder.*',
'^(?!TF|Flax).*Model$', '^(?!TF|Flax).*Adapter$', 'ImageProjection',
'^(?!TF|Flax).*UNet$'
]
if importlib.util.find_spec('transformers') is not None:
import transformers
lazy_module = sys.modules['transformers']