manbeast3b
/

deserve_edge1

Model card Files Files and versions

manbeast3b commited on Dec 15, 2024

Commit

f59ece9

·

verified ·

1 Parent(s): 9cc5040

Update src/pipeline.py

Files changed (1) hide show

src/pipeline.py +1 -1

src/pipeline.py CHANGED Viewed

@@ -370,7 +370,7 @@ class FluxTransformer2DModel(ModelMixin, ConfigMixin, PeftAdapterMixin, FromOrig
         self.norm_out = AdaLayerNormContinuous(self.inner_dim, self.inner_dim, elementwise_affine=False, eps=1e-6)
         self.proj_out = nn.Linear(self.inner_dim, patch_size * patch_size * self.out_channels, bias=True)
         ratio: float = 0.5
-        down: int = 1
         sx: int = 2
         sy: int = 2
         rando: bool = False

         self.norm_out = AdaLayerNormContinuous(self.inner_dim, self.inner_dim, elementwise_affine=False, eps=1e-6)
         self.proj_out = nn.Linear(self.inner_dim, patch_size * patch_size * self.out_channels, bias=True)
         ratio: float = 0.5
+        down: int = 3
         sx: int = 2
         sy: int = 2
         rando: bool = False