Skip to content

Commit b42bb54

Browse files
chenjy2003yiyixuxu
andauthored
remove reset_parameters for RMSNorm2d
Co-authored-by: YiYi Xu <[email protected]>
1 parent cab56b1 commit b42bb54

File tree

1 file changed

+0
-5
lines changed
  • src/diffusers/models/autoencoders

1 file changed

+0
-5
lines changed

src/diffusers/models/autoencoders/dc_ae.py

Lines changed: 0 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -52,11 +52,6 @@ def __init__(self, num_features: int, eps: float = 1e-5, elementwise_affine: boo
5252

5353
self.reset_parameters()
5454

55-
def reset_parameters(self) -> None:
56-
if self.elementwise_affine:
57-
torch.nn.init.ones_(self.weight)
58-
if self.bias is not None:
59-
torch.nn.init.zeros_(self.bias)
6055

6156
def forward(self, x: torch.Tensor) -> torch.Tensor:
6257
x = (x / torch.sqrt(torch.square(x.float()).mean(dim=1, keepdim=True) + self.eps)).to(x.dtype)

0 commit comments

Comments
 (0)