Signed-off-by: Yuan Tang <terrytangyuan@gmail.com>
This commit is contained in:
Yuan Tang 2025-03-21 11:27:55 -04:00
parent 7968148bca
commit 98274574f0
No known key found for this signature in database

View file

@ -207,6 +207,7 @@ class SamplingParams(BaseModel):
:param stop: Up to 4 sequences where the API will stop generating further tokens.
The returned text will not contain the stop sequence.
"""
strategy: SamplingStrategy = Field(default_factory=GreedySamplingStrategy)
max_tokens: Optional[int] = 0