Skip to content

GitLab

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

Fix ignore_eos (#1645)

parent 1d9deeac
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 2 deletions
python/sglang/srt/openai_api/protocol.py
View file @ e37cdab0
......@@ -170,8 +170,8 @@ class CompletionRequest(BaseModel):
# Extra parameters for SRT backend only and will be ignored by OpenAI models.
regex: Optional[str] = None
json_schema: Optional[str] = None
ignore_eos: Optional[bool] = False
min_tokens: Optional[int] = 0
ignore_eos: bool = False
min_tokens: int = 0
repetition_penalty: Optional[float] = 1.0
stop_token_ids: Optional[List[int]] = Field(default_factory=list)
......@@ -275,6 +275,7 @@ class ChatCompletionRequest(BaseModel):
min_tokens: Optional[int] = 0
repetition_penalty: Optional[float] = 1.0
stop_token_ids: Optional[List[int]] = Field(default_factory=list)
ignore_eos: bool = False
class ChatMessage(BaseModel):
......
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