mirror of
https://github.com/huggingface/text-generation-inference.git
synced 2025-09-11 20:34:54 +00:00
fix: adjust for idefics2 template
This commit is contained in:
parent
93409ea038
commit
dd696891ac
@ -101,7 +101,10 @@ impl Infer {
|
|||||||
})
|
})
|
||||||
.map(|t| {
|
.map(|t| {
|
||||||
// .strip() is not supported in minijinja
|
// .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)
|
ChatTemplate::new(t, tokenizer_config.bos_token, tokenizer_config.eos_token)
|
||||||
});
|
});
|
||||||
|
|
||||||
|
Loading…
Reference in New Issue
Block a user