4
4
from ..utils import DummyObject , requires_backends
5
5
6
6
7
- class VQDiffusionPipeline (metaclass = DummyObject ):
7
+ class LDMTextToImagePipeline (metaclass = DummyObject ):
8
8
_backends = ["torch" , "transformers" ]
9
9
10
10
def __init__ (self , * args , ** kwargs ):
@@ -19,7 +19,7 @@ def from_pretrained(cls, *args, **kwargs):
19
19
requires_backends (cls , ["torch" , "transformers" ])
20
20
21
21
22
- class LDMTextToImagePipeline (metaclass = DummyObject ):
22
+ class StableDiffusionImg2ImgPipeline (metaclass = DummyObject ):
23
23
_backends = ["torch" , "transformers" ]
24
24
25
25
def __init__ (self , * args , ** kwargs ):
@@ -34,7 +34,7 @@ def from_pretrained(cls, *args, **kwargs):
34
34
requires_backends (cls , ["torch" , "transformers" ])
35
35
36
36
37
- class StableDiffusionImg2ImgPipeline (metaclass = DummyObject ):
37
+ class StableDiffusionInpaintPipeline (metaclass = DummyObject ):
38
38
_backends = ["torch" , "transformers" ]
39
39
40
40
def __init__ (self , * args , ** kwargs ):
@@ -49,7 +49,7 @@ def from_pretrained(cls, *args, **kwargs):
49
49
requires_backends (cls , ["torch" , "transformers" ])
50
50
51
51
52
- class StableDiffusionInpaintPipeline (metaclass = DummyObject ):
52
+ class StableDiffusionInpaintPipelineLegacy (metaclass = DummyObject ):
53
53
_backends = ["torch" , "transformers" ]
54
54
55
55
def __init__ (self , * args , ** kwargs ):
@@ -64,7 +64,7 @@ def from_pretrained(cls, *args, **kwargs):
64
64
requires_backends (cls , ["torch" , "transformers" ])
65
65
66
66
67
- class StableDiffusionInpaintPipelineLegacy (metaclass = DummyObject ):
67
+ class StableDiffusionPipeline (metaclass = DummyObject ):
68
68
_backends = ["torch" , "transformers" ]
69
69
70
70
def __init__ (self , * args , ** kwargs ):
@@ -79,7 +79,7 @@ def from_pretrained(cls, *args, **kwargs):
79
79
requires_backends (cls , ["torch" , "transformers" ])
80
80
81
81
82
- class StableDiffusionPipeline (metaclass = DummyObject ):
82
+ class VQDiffusionPipeline (metaclass = DummyObject ):
83
83
_backends = ["torch" , "transformers" ]
84
84
85
85
def __init__ (self , * args , ** kwargs ):
0 commit comments