kanshan/diffusers
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

make style

Browse Source
This commit is contained in:
Patrick von Platen
2023-02-20 09:34:30 +02:00
parent 4135558a78
commit d7cfa0baa2
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/diffusers/pipelines/pipeline_utils.py
+2 -2
View File
@@ -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)