huggingface/text-generation-inference
5
0
Fork 0
mirror of https://github.com/huggingface/text-generation-inference.git synced 2025-09-11 12:24:53 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: revise temp scaling logic

Browse Source
This commit is contained in:
drbh 2024-04-15 20:13:40 +00:00
parent 24a5588735
commit 7879365f96
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
router/src/server.rs
View File

@ -1009,8 +1009,8 @@ async fn chat_completions(
let logprobs = logprobs.unwrap_or(false);
let tool_prompt = tool_prompt.unwrap_or_default();
let stop = stop.unwrap_or_default();
// rescale where 0 is deterministic and 1 is random (this is the opposite of other endpoints)
let adjusted_temperature = temperature.map_or(1.0, |t| 1.0 - t);
// rescale temperature starting from 0.0 to 1.0
let adjusted_temperature = req.temperature.map_or(1.0, |t| t + 1.0);
let do_sample = adjusted_temperature > 0.0;
let temperature = Some(adjusted_temperature);