squall/vllm
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

Fix __repr__ of SequenceOutputs (#1311)

Browse Source
This commit is contained in:
Wang Ran (汪然) 2023-10-11 00:58:28 +08:00 committed by GitHub
parent 6a6119554c
commit ac5cf86aa6
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
vllm/sequence.py
View File

@ -359,8 +359,8 @@ class SequenceOutputs:
def __repr__(self) -> str: def __repr__(self) -> str:
return (f"SequenceOutputs(parent_seq_id={self.parent_seq_id}, " return (f"SequenceOutputs(parent_seq_id={self.parent_seq_id}, "
f"output_token={self.output_token}), " f"output_token={self.output_token}, "
f"logprobs={self.logprobs}") f"logprobs={self.logprobs})")
def __eq__(self, other: object) -> bool: def __eq__(self, other: object) -> bool:
if not isinstance(other, SequenceOutputs): if not isinstance(other, SequenceOutputs):