Commit d7cfa0ba authored by Patrick von Platen's avatar Patrick von Platen
Browse files

make style

parent 4135558a
...@@ -825,12 +825,12 @@ class DiffusionPipeline(ConfigMixin): ...@@ -825,12 +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(): 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: else:
transformers_version = version.parse("not installed") 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