mirror of
https://github.com/PaddlePaddle/FastDeploy.git
synced 2025-10-05 16:48:03 +08:00
check (#3866)
This commit is contained in:
@@ -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
|
||||
|
Reference in New Issue
Block a user