Skip to content

Commit 92679c4

Browse files
haofanwangpatrickvonplaten
authored andcommitted
Update pipeline_utils.py (#2415)
1 parent a41b043 commit 92679c4

File tree

1 file changed

+6
-1
lines changed

1 file changed

+6
-1
lines changed

src/diffusers/pipelines/pipeline_utils.py

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -825,7 +825,12 @@ def load_module(name, value):
825825
loading_kwargs["sess_options"] = sess_options
826826

827827
is_diffusers_model = issubclass(class_obj, diffusers.ModelMixin)
828-
transformers_version = version.parse(version.parse(transformers.__version__).base_version)
828+
829+
if is_transformers_available():
830+
transformers_version = version.parse(version.parse(transformers.__version__).base_version)
831+
else:
832+
transformers_version = version.parse("not installed")
833+
829834
is_transformers_model = (
830835
is_transformers_available()
831836
and issubclass(class_obj, PreTrainedModel)

0 commit comments

Comments
 (0)