|
4 | 4 | from ..utils import DummyObject, requires_backends |
5 | 5 |
|
6 | 6 |
|
| 7 | +class OnnxStableDiffusionImg2ImgPipeline(metaclass=DummyObject): |
| 8 | + _backends = ["torch", "transformers", "onnx"] |
| 9 | + |
| 10 | + def __init__(self, *args, **kwargs): |
| 11 | + requires_backends(self, ["torch", "transformers", "onnx"]) |
| 12 | + |
| 13 | + @classmethod |
| 14 | + def from_config(cls, *args, **kwargs): |
| 15 | + requires_backends(cls, ["torch", "transformers", "onnx"]) |
| 16 | + |
| 17 | + @classmethod |
| 18 | + def from_pretrained(cls, *args, **kwargs): |
| 19 | + requires_backends(cls, ["torch", "transformers", "onnx"]) |
| 20 | + |
| 21 | + |
| 22 | +class OnnxStableDiffusionInpaintPipeline(metaclass=DummyObject): |
| 23 | + _backends = ["torch", "transformers", "onnx"] |
| 24 | + |
| 25 | + def __init__(self, *args, **kwargs): |
| 26 | + requires_backends(self, ["torch", "transformers", "onnx"]) |
| 27 | + |
| 28 | + @classmethod |
| 29 | + def from_config(cls, *args, **kwargs): |
| 30 | + requires_backends(cls, ["torch", "transformers", "onnx"]) |
| 31 | + |
| 32 | + @classmethod |
| 33 | + def from_pretrained(cls, *args, **kwargs): |
| 34 | + requires_backends(cls, ["torch", "transformers", "onnx"]) |
| 35 | + |
| 36 | + |
7 | 37 | class OnnxStableDiffusionPipeline(metaclass=DummyObject): |
8 | 38 | _backends = ["torch", "transformers", "onnx"] |
9 | 39 |
|
|
0 commit comments