make fix copies
This commit is contained in:
parent
8eb9d9703d
commit
bd216073fe
|
@ -4,6 +4,36 @@
|
|||
from ..utils import DummyObject, requires_backends
|
||||
|
||||
|
||||
class OnnxStableDiffusionImg2ImgPipeline(metaclass=DummyObject):
|
||||
_backends = ["torch", "transformers", "onnx"]
|
||||
|
||||
def __init__(self, *args, **kwargs):
|
||||
requires_backends(self, ["torch", "transformers", "onnx"])
|
||||
|
||||
@classmethod
|
||||
def from_config(cls, *args, **kwargs):
|
||||
requires_backends(cls, ["torch", "transformers", "onnx"])
|
||||
|
||||
@classmethod
|
||||
def from_pretrained(cls, *args, **kwargs):
|
||||
requires_backends(cls, ["torch", "transformers", "onnx"])
|
||||
|
||||
|
||||
class OnnxStableDiffusionInpaintPipeline(metaclass=DummyObject):
|
||||
_backends = ["torch", "transformers", "onnx"]
|
||||
|
||||
def __init__(self, *args, **kwargs):
|
||||
requires_backends(self, ["torch", "transformers", "onnx"])
|
||||
|
||||
@classmethod
|
||||
def from_config(cls, *args, **kwargs):
|
||||
requires_backends(cls, ["torch", "transformers", "onnx"])
|
||||
|
||||
@classmethod
|
||||
def from_pretrained(cls, *args, **kwargs):
|
||||
requires_backends(cls, ["torch", "transformers", "onnx"])
|
||||
|
||||
|
||||
class OnnxStableDiffusionPipeline(metaclass=DummyObject):
|
||||
_backends = ["torch", "transformers", "onnx"]
|
||||
|
||||
|
|
Loading…
Reference in New Issue