Skip to content

Commit bdb113c

Browse files
committed
refactor(tokenizer): drop intermediate Vec in Kimi-K2.5 arch detection
Signed-off-by: key4ng <rukeyang@gmail.com>
1 parent ce2fad1 commit bdb113c

1 file changed

Lines changed: 8 additions & 5 deletions

File tree

crates/tokenizer/src/tiktoken.rs

Lines changed: 8 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -594,11 +594,14 @@ fn detect_renderer_from_config(dir: &Path) -> Renderer {
594594
return Renderer::Jinja;
595595
}
596596
};
597-
let architectures = value.get("architectures").and_then(|v| v.as_array());
598-
let arch_strs: Vec<&str> = architectures
599-
.map(|a| a.iter().filter_map(|v| v.as_str()).collect())
600-
.unwrap_or_default();
601-
if arch_strs.contains(&"KimiK25ForConditionalGeneration") {
597+
let is_kimi = value
598+
.get("architectures")
599+
.and_then(|v| v.as_array())
600+
.is_some_and(|a| {
601+
a.iter()
602+
.any(|v| v.as_str() == Some("KimiK25ForConditionalGeneration"))
603+
});
604+
if is_kimi {
602605
tracing::debug!(?path, "selected KimiK25Tools chat-template renderer");
603606
return Renderer::KimiK25Tools;
604607
}

0 commit comments

Comments
 (0)