mirror of
https://git.datalinker.icu/vllm-project/vllm.git
synced 2025-12-09 13:15:48 +08:00
Add find method to sequence group
This commit is contained in:
parent
eb52db1bea
commit
0961f5a49a
@ -76,3 +76,9 @@ class SequenceGroup:
|
||||
return len(self.seqs)
|
||||
else:
|
||||
return len([seq for seq in self.seqs if seq.status == status])
|
||||
|
||||
def find(self, seq_id: int) -> Sequence:
|
||||
for seq in self.seqs:
|
||||
if seq.seq_id == seq_id:
|
||||
return seq
|
||||
raise ValueError(f'Sequence {seq_id} not found.')
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user