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

Fix ignore_eos (#1645)

Browse Source
This commit is contained in:
Lianmin Zheng
2024-10-12 00:36:28 -07:00
committed by GitHub
parent 1d9deeacdb
commit e37cdab0c6
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
python/sglang/srt/openai_api/protocol.py
View File

@@ -170,8 +170,8 @@ class CompletionRequest(BaseModel):
# Extra parameters for SRT backend only and will be ignored by OpenAI models. # Extra parameters for SRT backend only and will be ignored by OpenAI models.
regex: Optional[str] = None regex: Optional[str] = None
json_schema: Optional[str] = None json_schema: Optional[str] = None
ignore_eos: Optional[bool] = False ignore_eos: bool = False
min_tokens: Optional[int] = 0 min_tokens: int = 0
repetition_penalty: Optional[float] = 1.0 repetition_penalty: Optional[float] = 1.0
stop_token_ids: Optional[List[int]] = Field(default_factory=list) stop_token_ids: Optional[List[int]] = Field(default_factory=list)
@@ -275,6 +275,7 @@ class ChatCompletionRequest(BaseModel):
min_tokens: Optional[int] = 0 min_tokens: Optional[int] = 0
repetition_penalty: Optional[float] = 1.0 repetition_penalty: Optional[float] = 1.0
stop_token_ids: Optional[List[int]] = Field(default_factory=list) stop_token_ids: Optional[List[int]] = Field(default_factory=list)
ignore_eos: bool = False
class ChatMessage(BaseModel): class ChatMessage(BaseModel):