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

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)