mirror of
https://github.com/meta-llama/llama-stack.git
synced 2025-12-29 11:28:48 +00:00
updated the additional params to pass any type of values
This commit is contained in:
parent
1d6ef73dd7
commit
03a25a7753
3 changed files with 28 additions and 3 deletions
|
|
@ -82,7 +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]] = {}
|
||||
additional_params: Optional[Dict[str, Any]] = {}
|
||||
|
||||
|
||||
class LogProbConfig(BaseModel):
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue