apps/FastDeploy
1
0
Fork 0
mirror of https://github.com/PaddlePaddle/FastDeploy.git synced 2025-10-07 01:22:59 +08:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

check (#3866)

Browse Source
This commit is contained in:
chen
2025-09-03 20:46:13 +08:00
committed by GitHub
parent cc5430e4c2
commit c2f5c99b1e
1 changed files with 0 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
fastdeploy/model_executor/layers/lm_head.py
View File

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