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:
kingbri 2024-02-07 21:41:15 -05:00 committed by Brian Dashore
parent 2642ef7156
commit 0af6a38af3
6 changed files with 145 additions and 52 deletions

View file

@ -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