Skip to content

Commit f6a41db

Browse files
committed
make fix-copies.
1 parent fb99712 commit f6a41db

File tree

1 file changed

+15
-0
lines changed

1 file changed

+15
-0
lines changed

src/diffusers/utils/dummy_torch_and_transformers_objects.py

Lines changed: 15 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1622,6 +1622,21 @@ def from_pretrained(cls, *args, **kwargs):
16221622
requires_backends(cls, ["torch", "transformers"])
16231623

16241624

1625+
class SanaSprintImg2ImgPipeline(metaclass=DummyObject):
1626+
_backends = ["torch", "transformers"]
1627+
1628+
def __init__(self, *args, **kwargs):
1629+
requires_backends(self, ["torch", "transformers"])
1630+
1631+
@classmethod
1632+
def from_config(cls, *args, **kwargs):
1633+
requires_backends(cls, ["torch", "transformers"])
1634+
1635+
@classmethod
1636+
def from_pretrained(cls, *args, **kwargs):
1637+
requires_backends(cls, ["torch", "transformers"])
1638+
1639+
16251640
class SanaSprintPipeline(metaclass=DummyObject):
16261641
_backends = ["torch", "transformers"]
16271642

0 commit comments

Comments
 (0)