Merge pull request #556 from XxAlonexX/main

Fix Linear Layer Bias Initialization
This commit is contained in:
Xingkai Yu 2025-02-05 16:23:02 +08:00 committed by GitHub
commit 87a01053e4
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -185,7 +185,7 @@ class Linear(nn.Module):
else:
self.register_parameter("scale", None)
if bias:
self.bias = nn.Parameter(torch.empty(self.part_out_features))
self.bias = nn.Parameter(torch.empty(out_features))
else:
self.register_parameter("bias", None)