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: make logic more readable

Browse Source
This commit is contained in:
drbh 2024-04-17 20:20:35 +00:00
parent 9906b03b60
commit 7ecda467c2
1 changed files with 2 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
router/src/server.rs
View File

@ -1010,9 +1010,8 @@ async fn chat_completions(
let tool_prompt = tool_prompt.unwrap_or_default();
let stop = stop.unwrap_or_default();
// enable greedy only when temperature is 0
let do_sample = temperature.map_or(true, |t| t > 0.0);
let adjusted_temperature = temperature.map_or(1.0, |t| if t == 0.0 { 1.0 } else { t });
let temperature = Some(adjusted_temperature);
let do_sample = temperature.map_or(true, |t| t != 0.0);
let temperature = temperature.map(|t| if t == 0.0 { 1.0 } else { t });
// extract tool grammar if present
let tool_grammar = match ToolGrammar::apply(tools, tool_choice) {