This commit is contained in:
chen
2025-09-03 20:46:13 +08:00
committed by GitHub
parent cc5430e4c2
commit c2f5c99b1e

View File

@@ -140,5 +140,4 @@ class ParallelLMHead(nn.Layer):
logits = paddle.matmul(logits.astype(self.weight.dtype), self.weight)
else:
logits = self.linear(logits.astype(self.linear.weight.dtype))
print(self.linear.weight.dtype)
return logits