Skip to content

GitLab

Unverified Commit 5dd0f870 authored by Simo Lin's avatar Simo Lin Committed by GitHub
Browse files

[bug] fix pd completion protocol for batching support (#8317)

parent f7e102d5
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 3 deletions
python/sglang/srt/entrypoints/openai/protocol.py
View file @ 5dd0f870
...@@ -192,9 +192,9 @@ class CompletionRequest(BaseModel): ...@@ -192,9 +192,9 @@ class CompletionRequest(BaseModel):
session_params: Optional[Dict] = None session_params: Optional[Dict] = None
# For PD disaggregation # For PD disaggregation
bootstrap_host: Optional[str] = None bootstrap_host: Optional[Union[List[str], str]] = None
bootstrap_port: Optional[int] = None bootstrap_port: Optional[Union[List[Optional[int]], int]] = None
bootstrap_room: Optional[int] = None bootstrap_room: Optional[Union[List[int], int]] = None
# For request id # For request id
rid: Optional[Union[List[str], str]] = None rid: Optional[Union[List[str], str]] = None
......
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