EngineX-Hygon/sglang
5
0
Fork 0
Code Issues Pull Requests Actions 7 Projects Releases Wiki Activity

Fix MockTokenizer in the unit tests (#1774)

Browse Source
This commit is contained in:
Lianmin Zheng
2024-10-23 17:47:05 -07:00
committed by GitHub
parent 8f8f96a621
commit 87a7cfa080
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
python/sglang/test/srt/sampling/penaltylib/utils.py
View File

@@ -24,6 +24,7 @@ class MockSamplingParams:
@dataclasses.dataclass @dataclasses.dataclass
class MockTokenizer: class MockTokenizer:
eos_token_id: int eos_token_id: int
additional_stop_token_ids: typing.Optional[typing.List[int]] = None
@dataclasses.dataclass @dataclasses.dataclass