OAI: Add stream to gen params
Good for logging. Signed-off-by: kingbri <bdashore3@proton.me>
This commit is contained in:
parent
69e41e994c
commit
56fdfb5f8e
2 changed files with 5 additions and 1 deletions
|
|
@ -863,6 +863,7 @@ class ExllamaV2Container:
|
|||
# Some options are too large, so log the args instead
|
||||
log_generation_params(
|
||||
max_tokens=max_tokens,
|
||||
stream=kwargs.get("stream"),
|
||||
**gen_settings_log_dict,
|
||||
token_healing=token_healing,
|
||||
auto_scale_penalty_range=auto_scale_penalty_range,
|
||||
|
|
|
|||
|
|
@ -43,6 +43,9 @@ class CommonCompletionRequest(BaseSamplerRequest):
|
|||
)
|
||||
|
||||
def to_gen_params(self):
|
||||
extra_gen_params = {"logprobs": self.logprobs}
|
||||
extra_gen_params = {
|
||||
"stream": self.stream,
|
||||
"logprobs": self.logprobs,
|
||||
}
|
||||
|
||||
return super().to_gen_params(**extra_gen_params)
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue