mirror of
https://github.com/meta-llama/llama-stack.git
synced 2025-12-28 18:42:00 +00:00
added additional params and new functions required to watsonx
This commit is contained in:
parent
1bb1d9b2ba
commit
1d6ef73dd7
4 changed files with 148 additions and 2 deletions
|
|
@ -82,6 +82,7 @@ class SamplingParams(BaseModel):
|
|||
max_tokens: Optional[int] = 0
|
||||
repetition_penalty: Optional[float] = 1.0
|
||||
stop: Optional[List[str]] = None
|
||||
additional_params: Optional[Dict[str, str]] = {}
|
||||
|
||||
|
||||
class LogProbConfig(BaseModel):
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue