AI/diffusers
1
0
Fork 0
mirror of https://github.com/huggingface/diffusers.git synced 2026-01-27 17:22:53 +03:00
Code Activity

Update pipeline_utils.py (#2415)

Browse Source
This commit is contained in:
Haofan Wang
2023-02-20 15:34:13 +08:00
committed by Patrick von Platen
parent a41b043570
commit 92679c4851
1 changed files with 6 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
src/diffusers/pipelines/pipeline_utils.py
View File

@@ -825,7 +825,12 @@ class DiffusionPipeline(ConfigMixin):
loading_kwargs["sess_options"] = sess_options
is_diffusers_model = issubclass(class_obj, diffusers.ModelMixin)
transformers_version = version.parse(version.parse(transformers.__version__).base_version)
if is_transformers_available():
transformers_version = version.parse(version.parse(transformers.__version__).base_version)
else:
transformers_version = version.parse("not installed")
is_transformers_model = (
is_transformers_available()
and issubclass(class_obj, PreTrainedModel)