mirror of
https://github.com/theroyallab/tabbyAPI.git
synced 2026-03-15 00:07:28 +00:00
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:
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user