Skip to content

Commit bd21607

Browse files
make fix copies
1 parent 8eb9d97 commit bd21607

File tree

1 file changed

+30
-0
lines changed

1 file changed

+30
-0
lines changed

src/diffusers/utils/dummy_torch_and_transformers_and_onnx_objects.py

Lines changed: 30 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -4,6 +4,36 @@
44
from ..utils import DummyObject, requires_backends
55

66

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+
737
class OnnxStableDiffusionPipeline(metaclass=DummyObject):
838
_backends = ["torch", "transformers", "onnx"]
939

0 commit comments

Comments
 (0)