quickjkee commited on
Commit
8f65a53
·
verified ·
1 Parent(s): 05a4087

Update pipeline.py

Browse files
Files changed (1) hide show
  1. 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__(