Update pipeline.py
Browse files- pipeline.py +2 -0
pipeline.py
CHANGED
@@ -50,6 +50,7 @@ else:
|
|
50 |
|
51 |
class SwDPipeline(StableDiffusion3Pipeline):
|
52 |
|
|
|
53 |
def __init__(
|
54 |
self,
|
55 |
transformer: SD3Transformer2DModel,
|
@@ -92,6 +93,7 @@ class SwDPipeline(StableDiffusion3Pipeline):
|
|
92 |
self.patch_size = (
|
93 |
self.transformer.config.patch_size if hasattr(self, "transformer") and self.transformer is not None else 2
|
94 |
)
|
|
|
95 |
|
96 |
@torch.no_grad()
|
97 |
def __call__(
|
|
|
50 |
|
51 |
class SwDPipeline(StableDiffusion3Pipeline):
|
52 |
|
53 |
+
"""
|
54 |
def __init__(
|
55 |
self,
|
56 |
transformer: SD3Transformer2DModel,
|
|
|
93 |
self.patch_size = (
|
94 |
self.transformer.config.patch_size if hasattr(self, "transformer") and self.transformer is not None else 2
|
95 |
)
|
96 |
+
"""
|
97 |
|
98 |
@torch.no_grad()
|
99 |
def __call__(
|