We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent b42bb54 commit 2e04a99Copy full SHA for 2e04a99
src/diffusers/models/autoencoders/dc_ae.py
@@ -34,7 +34,7 @@
34
35
36
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:
+ def __init__(self, num_features: int, eps: float = 1e-5, elementwise_affine: bool = True, bias: bool = True) -> None:
38
factory_kwargs = {'device': device, 'dtype': dtype}
39
super().__init__()
40
self.num_features = num_features
0 commit comments