Update pipeline_utils.py (#2415)

4135558a · Haofan Wang · GitHub · 45572c24 · 4135558a
Unverified Commit 4135558a authored Feb 20, 2023 by Haofan Wang Committed by GitHub Feb 20, 2023
Hide whitespace changes
Inline Side-by-side

Showing with 6 additions and 1 deletion

src/diffusers/pipelines/pipeline_utils.py src/diffusers/pipelines/pipeline_utils.py +6 -1

No files found.
--- a/src/diffusers/pipelines/pipeline_utils.py
+++ b/src/diffusers/pipelines/pipeline_utils.py
@@ -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)