Skip to content

Commit a24caae

Browse files
thehamstinoaccOS
authored andcommitted
ollama: Add max tokens for qwen2.5-coder (zed-industries#18290)
1 parent 609b4d4 commit a24caae

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

crates/ollama/src/ollama.rs

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -82,7 +82,8 @@ fn get_max_tokens(name: &str) -> usize {
8282
"llama3" | "gemma2" | "gemma" | "codegemma" | "starcoder" | "aya" => 8192,
8383
"codellama" | "starcoder2" => 16384,
8484
"mistral" | "codestral" | "mixstral" | "llava" | "qwen2" | "dolphin-mixtral" => 32768,
85-
"llama3.1" | "phi3" | "phi3.5" | "command-r" | "deepseek-coder-v2" | "yi-coder" => 128000,
85+
"llama3.1" | "phi3" | "phi3.5" | "command-r" | "deepseek-coder-v2" | "yi-coder"
86+
| "qwen2.5-coder" => 128000,
8687
_ => DEFAULT_TOKENS,
8788
}
8889
.clamp(1, MAXIMUM_TOKENS)

0 commit comments

Comments
 (0)