API: Add temperature_last support

Documented in previous commits. Also make sure that for version checking,
check the value of kwargs instead of if the key is present since requests
pass default values.

Signed-off-by: kingbri <bdashore3@proton.me>
This commit is contained in:
kingbri
2023-11-21 21:20:59 -05:00
parent 3337fe6acc
commit 71b9a53336
2 changed files with 6 additions and 4 deletions

View File

@@ -40,6 +40,7 @@ class CommonCompletionRequest(BaseModel):
# Sampling params
token_healing: Optional[bool] = False
temperature: Optional[float] = 1.0
temperature_last: Optional[bool] = False
top_k: Optional[int] = 0
top_p: Optional[float] = 1.0
typical: Optional[float] = 0.0
@@ -71,6 +72,7 @@ class CommonCompletionRequest(BaseModel):
"ban_eos_token": self.ban_eos_token,
"token_healing": self.token_healing,
"temperature": self.temperature,
"temperature_last": self.temperature_last,
"top_k": self.top_k,
"top_p": self.top_p,
"typical": self.typical,
@@ -81,5 +83,5 @@ class CommonCompletionRequest(BaseModel):
"repetition_decay": self.repetition_decay,
"mirostat": self.mirostat_mode == 2,
"mirostat_tau": self.mirostat_tau,
"mirostat_eta": self.mirostat_eta
"mirostat_eta": self.mirostat_eta,
}