Skip to content

Commit cf75374

Browse files
authored
Fix logprobs to_tensor(#4132)
1 parent e1d4a47 commit cf75374

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

lmdeploy/pytorch/engine/model_agent.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -58,7 +58,7 @@ def to_tensor(self):
5858
if isinstance(self.vals, torch.Tensor):
5959
vals = self.vals
6060
else:
61-
vals = torch.from_numpy(vals)
61+
vals = torch.from_numpy(self.vals)
6262
return BatchedLogProbs(vals=vals, indices=torch.from_numpy(self.indices))
6363

6464

0 commit comments

Comments
 (0)