SeedVR2-3B-Image-Upscale

Running on Zero

IceClear commited on Jun 17

Commit

39245bd

1 Parent(s): 96af013

update

Files changed (1) hide show

models/dit_v2/normalization.py CHANGED Viewed

@@ -30,7 +30,7 @@ def get_norm_layer(norm_type: Optional[str]) -> norm_layer_type:
             return nn.LayerNorm(
                 normalized_shape=dim,
                 eps=eps,
-                elementwise_affine=elementwise_affine,
             )
         if norm_type == "rms":

             return nn.LayerNorm(
                 normalized_shape=dim,
                 eps=eps,
+                elementwise_affine=False,
             )
         if norm_type == "rms":