diff --git a/router/src/infer.rs b/router/src/infer.rs index 1123dd6a..1447e756 100644 --- a/router/src/infer.rs +++ b/router/src/infer.rs @@ -101,7 +101,10 @@ impl Infer { }) .map(|t| { // .strip() is not supported in minijinja - let t = t.replace(".strip()", " | trim"); + // .capitalize() is not supported in minijinja but we can use | capitalize + let t = t + .replace(".strip()", " | trim") + .replace(".capitalize()", " | capitalize"); ChatTemplate::new(t, tokenizer_config.bos_token, tokenizer_config.eos_token) });