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 c82f828 commit 22ea5fdCopy full SHA for 22ea5fd
src/diffusers/models/autoencoders/dc_ae.py
@@ -42,7 +42,7 @@ def __init__(self, num_features: int, eps: float = 1e-5, elementwise_affine: boo
42
if self.elementwise_affine:
43
self.weight = torch.nn.parameter.Parameter(torch.empty(self.num_features))
44
if bias:
45
- self.bias = torch.nn.parameter.Parameter(torch.empty(self.num_features, **factory_kwargs))
+ self.bias = torch.nn.parameter.Parameter(torch.empty(self.num_features))
46
else:
47
self.register_parameter('bias', None)
48
0 commit comments