fix copies

Author: patrickvonplaten

src/diffusers/utils/dummy_torch_and_accelerate_objects.py

@@ -34,6 +34,21 @@ def from_pretrained(cls, *args, **kwargs):
         requires_backends(cls, ["torch", "accelerate"])
 
 
+class Transformer2DModel(metaclass=DummyObject):
+    _backends = ["torch", "accelerate"]
+
+    def __init__(self, *args, **kwargs):
+        requires_backends(self, ["torch", "accelerate"])
+
+    @classmethod
+    def from_config(cls, *args, **kwargs):
+        requires_backends(cls, ["torch", "accelerate"])
+
+    @classmethod
+    def from_pretrained(cls, *args, **kwargs):
+        requires_backends(cls, ["torch", "accelerate"])
+
+
 class UNet1DModel(metaclass=DummyObject):
     _backends = ["torch", "accelerate"]
 
@@ -227,6 +242,21 @@ def from_pretrained(cls, *args, **kwargs):
         requires_backends(cls, ["torch", "accelerate"])
 
 
+class RePaintPipeline(metaclass=DummyObject):
+    _backends = ["torch", "accelerate"]
+
+    def __init__(self, *args, **kwargs):
+        requires_backends(self, ["torch", "accelerate"])
+
+    @classmethod
+    def from_config(cls, *args, **kwargs):
+        requires_backends(cls, ["torch", "accelerate"])
+
+    @classmethod
+    def from_pretrained(cls, *args, **kwargs):
+        requires_backends(cls, ["torch", "accelerate"])
+
+
 class ScoreSdeVePipeline(metaclass=DummyObject):
     _backends = ["torch", "accelerate"]
 
@@ -242,6 +272,21 @@ def from_pretrained(cls, *args, **kwargs):
         requires_backends(cls, ["torch", "accelerate"])
 
 
+class VQDiffusionPipeline(metaclass=DummyObject):
+    _backends = ["torch", "accelerate"]
+
+    def __init__(self, *args, **kwargs):
+        requires_backends(self, ["torch", "accelerate"])
+
+    @classmethod
+    def from_config(cls, *args, **kwargs):
+        requires_backends(cls, ["torch", "accelerate"])
+
+    @classmethod
+    def from_pretrained(cls, *args, **kwargs):
+        requires_backends(cls, ["torch", "accelerate"])
+
+
 class DDIMScheduler(metaclass=DummyObject):
     _backends = ["torch", "accelerate"]
 
@@ -347,6 +392,21 @@ def from_pretrained(cls, *args, **kwargs):
         requires_backends(cls, ["torch", "accelerate"])
 
 
+class RePaintScheduler(metaclass=DummyObject):
+    _backends = ["torch", "accelerate"]
+
+    def __init__(self, *args, **kwargs):
+        requires_backends(self, ["torch", "accelerate"])
+
+    @classmethod
+    def from_config(cls, *args, **kwargs):
+        requires_backends(cls, ["torch", "accelerate"])
+
+    @classmethod
+    def from_pretrained(cls, *args, **kwargs):
+        requires_backends(cls, ["torch", "accelerate"])
+
+
 class SchedulerMixin(metaclass=DummyObject):
     _backends = ["torch", "accelerate"]
 
@@ -377,6 +437,21 @@ def from_pretrained(cls, *args, **kwargs):
         requires_backends(cls, ["torch", "accelerate"])
 
 
+class VQDiffusionScheduler(metaclass=DummyObject):
+    _backends = ["torch", "accelerate"]
+
+    def __init__(self, *args, **kwargs):
+        requires_backends(self, ["torch", "accelerate"])
+
+    @classmethod
+    def from_config(cls, *args, **kwargs):
+        requires_backends(cls, ["torch", "accelerate"])
+
+    @classmethod
+    def from_pretrained(cls, *args, **kwargs):
+        requires_backends(cls, ["torch", "accelerate"])
+
+
 class EMAModel(metaclass=DummyObject):
     _backends = ["torch", "accelerate"]