Skip to content

Commit be29631

Browse files
committed
fixes from "make fix-copies"
1 parent 92671ab commit be29631

File tree

2 files changed

+30
-0
lines changed

2 files changed

+30
-0
lines changed

src/diffusers/utils/dummy_pt_objects.py

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

500500

501+
class BriaTransformer2DModel(metaclass=DummyObject):
502+
_backends = ["torch"]
503+
504+
def __init__(self, *args, **kwargs):
505+
requires_backends(self, ["torch"])
506+
507+
@classmethod
508+
def from_config(cls, *args, **kwargs):
509+
requires_backends(cls, ["torch"])
510+
511+
@classmethod
512+
def from_pretrained(cls, *args, **kwargs):
513+
requires_backends(cls, ["torch"])
514+
515+
501516
class CacheMixin(metaclass=DummyObject):
502517
_backends = ["torch"]
503518

src/diffusers/utils/dummy_torch_and_transformers_objects.py

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

334334

335+
class BriaPipeline(metaclass=DummyObject):
336+
_backends = ["torch", "transformers"]
337+
338+
def __init__(self, *args, **kwargs):
339+
requires_backends(self, ["torch", "transformers"])
340+
341+
@classmethod
342+
def from_config(cls, *args, **kwargs):
343+
requires_backends(cls, ["torch", "transformers"])
344+
345+
@classmethod
346+
def from_pretrained(cls, *args, **kwargs):
347+
requires_backends(cls, ["torch", "transformers"])
348+
349+
335350
class ChromaImg2ImgPipeline(metaclass=DummyObject):
336351
_backends = ["torch", "transformers"]
337352

0 commit comments

Comments
 (0)