From 0b1a843d321fe08343c738dab4bf01e81520c16b Mon Sep 17 00:00:00 2001 From: Patrick von Platen Date: Wed, 31 Aug 2022 18:54:36 +0200 Subject: [PATCH] Check dummy file (#297) fix line type --- .../utils/dummy_transformers_objects.py | 28 +++++++++++++++++++ utils/check_dummies.py | 4 +-- 2 files changed, 30 insertions(+), 2 deletions(-) diff --git a/src/diffusers/utils/dummy_transformers_objects.py b/src/diffusers/utils/dummy_transformers_objects.py index 753e3fdbe2..4c216b6248 100644 --- a/src/diffusers/utils/dummy_transformers_objects.py +++ b/src/diffusers/utils/dummy_transformers_objects.py @@ -1,3 +1,31 @@ # This file is autogenerated by the command `make fix-copies`, do not edit. # flake8: noqa from ..utils import DummyObject, requires_backends + + +class LDMTextToImagePipeline(metaclass=DummyObject): + _backends = ["transformers"] + + def __init__(self, *args, **kwargs): + requires_backends(self, ["transformers"]) + + +class StableDiffusionImg2ImgPipeline(metaclass=DummyObject): + _backends = ["transformers"] + + def __init__(self, *args, **kwargs): + requires_backends(self, ["transformers"]) + + +class StableDiffusionInpaintPipeline(metaclass=DummyObject): + _backends = ["transformers"] + + def __init__(self, *args, **kwargs): + requires_backends(self, ["transformers"]) + + +class StableDiffusionPipeline(metaclass=DummyObject): + _backends = ["transformers"] + + def __init__(self, *args, **kwargs): + requires_backends(self, ["transformers"]) diff --git a/utils/check_dummies.py b/utils/check_dummies.py index f9a45284f3..928f9ac18a 100644 --- a/utils/check_dummies.py +++ b/utils/check_dummies.py @@ -77,8 +77,8 @@ def read_init(): single_line_import_search = _re_single_line_import.search(line) if single_line_import_search is not None: objects.extend(single_line_import_search.groups()[0].split(", ")) - elif line.startswith(" " * 12): - objects.append(line[12:-2]) + elif line.startswith(" " * 8): + objects.append(line[8:-2]) line_index += 1 backend_specific_objects[backend] = objects