"git@developer.sourcefind.cn:norm/vllm.git" did not exist on "add055e151f32f89dab5932d25e5285b2fc823f1"
Commit 988c8222 authored by Patrick von Platen's avatar Patrick von Platen
Browse files

fix copies

parent 74821781
...@@ -34,6 +34,21 @@ class AutoencoderKL(metaclass=DummyObject): ...@@ -34,6 +34,21 @@ class AutoencoderKL(metaclass=DummyObject):
requires_backends(cls, ["torch", "accelerate"]) 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): class UNet1DModel(metaclass=DummyObject):
_backends = ["torch", "accelerate"] _backends = ["torch", "accelerate"]
...@@ -227,6 +242,21 @@ class PNDMPipeline(metaclass=DummyObject): ...@@ -227,6 +242,21 @@ class PNDMPipeline(metaclass=DummyObject):
requires_backends(cls, ["torch", "accelerate"]) 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): class ScoreSdeVePipeline(metaclass=DummyObject):
_backends = ["torch", "accelerate"] _backends = ["torch", "accelerate"]
...@@ -242,6 +272,21 @@ class ScoreSdeVePipeline(metaclass=DummyObject): ...@@ -242,6 +272,21 @@ class ScoreSdeVePipeline(metaclass=DummyObject):
requires_backends(cls, ["torch", "accelerate"]) 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): class DDIMScheduler(metaclass=DummyObject):
_backends = ["torch", "accelerate"] _backends = ["torch", "accelerate"]
...@@ -347,6 +392,21 @@ class PNDMScheduler(metaclass=DummyObject): ...@@ -347,6 +392,21 @@ class PNDMScheduler(metaclass=DummyObject):
requires_backends(cls, ["torch", "accelerate"]) 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): class SchedulerMixin(metaclass=DummyObject):
_backends = ["torch", "accelerate"] _backends = ["torch", "accelerate"]
...@@ -377,6 +437,21 @@ class ScoreSdeVeScheduler(metaclass=DummyObject): ...@@ -377,6 +437,21 @@ class ScoreSdeVeScheduler(metaclass=DummyObject):
requires_backends(cls, ["torch", "accelerate"]) 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): class EMAModel(metaclass=DummyObject):
_backends = ["torch", "accelerate"] _backends = ["torch", "accelerate"]
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment