Skip to content

Commit 2e04a99

Browse files
chenjy2003yiyixuxu
andauthored
remove device and dtype in RMSNorm2d __init__
Co-authored-by: YiYi Xu <[email protected]>
1 parent b42bb54 commit 2e04a99

File tree

1 file changed

+1
-1
lines changed
  • src/diffusers/models/autoencoders

1 file changed

+1
-1
lines changed

src/diffusers/models/autoencoders/dc_ae.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -34,7 +34,7 @@
3434

3535

3636
class RMSNorm2d(nn.Module):
37-
def __init__(self, num_features: int, eps: float = 1e-5, elementwise_affine: bool = True, bias: bool = True, device=None, dtype=None) -> None:
37+
def __init__(self, num_features: int, eps: float = 1e-5, elementwise_affine: bool = True, bias: bool = True) -> None:
3838
factory_kwargs = {'device': device, 'dtype': dtype}
3939
super().__init__()
4040
self.num_features = num_features

0 commit comments

Comments
 (0)