Skip to content

GitLab

Unverified Commit 87a7cfa0 authored by Lianmin Zheng's avatar Lianmin Zheng Committed by GitHub
Browse files

Fix MockTokenizer in the unit tests (#1774)

parent 8f8f96a6
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
python/sglang/test/srt/sampling/penaltylib/utils.py
View file @ 87a7cfa0
...@@ -24,6 +24,7 @@ class MockSamplingParams: ...@@ -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
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment