Model: Add logprobs support
Returns token offsets, selected tokens, probabilities of tokens post-sampling, and normalized probability of selecting a token pre-sampling (for efficiency purposes). Only for text completions. Chat completions in a later commit. Signed-off-by: kingbri <bdashore3@proton.me>
This commit is contained in:
parent
2642ef7156
commit
0af6a38af3
6 changed files with 145 additions and 52 deletions
|
|
@ -159,7 +159,7 @@ class BaseSamplerRequest(BaseModel):
|
|||
examples=[1.0],
|
||||
)
|
||||
|
||||
def to_gen_params(self):
|
||||
def to_gen_params(self, **kwargs):
|
||||
"""Converts samplers to internal generation params"""
|
||||
|
||||
# Add forced overrides if present
|
||||
|
|
@ -201,7 +201,7 @@ class BaseSamplerRequest(BaseModel):
|
|||
"negative_prompt": self.negative_prompt,
|
||||
}
|
||||
|
||||
return gen_params
|
||||
return {**gen_params, **kwargs}
|
||||
|
||||
|
||||
# Global for default overrides
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue