Unverified Commit 4135558a authored by Haofan Wang's avatar Haofan Wang Committed by GitHub
Browse files

Update pipeline_utils.py (#2415)

parent 45572c24
...@@ -825,7 +825,12 @@ class DiffusionPipeline(ConfigMixin): ...@@ -825,7 +825,12 @@ class DiffusionPipeline(ConfigMixin):
loading_kwargs["sess_options"] = sess_options loading_kwargs["sess_options"] = sess_options
is_diffusers_model = issubclass(class_obj, diffusers.ModelMixin) is_diffusers_model = issubclass(class_obj, diffusers.ModelMixin)
if is_transformers_available():
transformers_version = version.parse(version.parse(transformers.__version__).base_version) transformers_version = version.parse(version.parse(transformers.__version__).base_version)
else:
transformers_version = version.parse("not installed")
is_transformers_model = ( is_transformers_model = (
is_transformers_available() is_transformers_available()
and issubclass(class_obj, PreTrainedModel) and issubclass(class_obj, PreTrainedModel)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment