"docs/source/en/api/pipelines/latent_diffusion.md" did not exist on "a6e2c1fe5c02cae8a9f077f5d4e11b73d5791723"
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
......@@ -192,9 +192,9 @@ class CompletionRequest(BaseModel):
session_params: Optional[Dict] = None
# For PD disaggregation
bootstrap_host: Optional[str] = None
bootstrap_port: Optional[int] = None
bootstrap_room: Optional[int] = None
bootstrap_host: Optional[Union[List[str], str]] = None
bootstrap_port: Optional[Union[List[Optional[int]], int]] = None
bootstrap_room: Optional[Union[List[int], int]] = None
# For request id
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