Skip to content

Commit 22ea5fd

Browse files
Update src/diffusers/models/autoencoders/dc_ae.py
Co-authored-by: YiYi Xu <[email protected]>
1 parent c82f828 commit 22ea5fd

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
@@ -42,7 +42,7 @@ def __init__(self, num_features: int, eps: float = 1e-5, elementwise_affine: boo
4242
if self.elementwise_affine:
4343
self.weight = torch.nn.parameter.Parameter(torch.empty(self.num_features))
4444
if bias:
45-
self.bias = torch.nn.parameter.Parameter(torch.empty(self.num_features, **factory_kwargs))
45+
self.bias = torch.nn.parameter.Parameter(torch.empty(self.num_features))
4646
else:
4747
self.register_parameter('bias', None)
4848
else:

0 commit comments

Comments
 (0)