mirrors/zed
1
0
Fork 0
mirror of https://github.com/zed-industries/zed.git synced 2025-01-24 02:46:43 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 12

ollama: Add deepseek-r1 context size to defaults (#23420)

Browse source
This commit is contained in:
Peter Tripp 2025-01-21 13:31:15 -05:00 committed by GitHub
parent aa5fa4b7d4
commit 14cd178ab0
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
crates/ollama/src/ollama.rs
View file

@ -84,7 +84,7 @@ fn get_max_tokens(name: &str) -> usize {
"mistral" | "codestral" | "mixstral" | "llava" | "qwen2" | "qwen2.5-coder" "mistral" | "codestral" | "mixstral" | "llava" | "qwen2" | "qwen2.5-coder"
| "dolphin-mixtral" => 32768, | "dolphin-mixtral" => 32768,
"llama3.1" | "phi3" | "phi3.5" | "phi4" | "command-r" | "deepseek-coder-v2" "llama3.1" | "phi3" | "phi3.5" | "phi4" | "command-r" | "deepseek-coder-v2"
| "yi-coder" | "llama3.2" => 128000, | "deepseek-r1" | "yi-coder" | "llama3.2" => 128000,
_ => DEFAULT_TOKENS, _ => DEFAULT_TOKENS,
} }
.clamp(1, MAXIMUM_TOKENS) .clamp(1, MAXIMUM_TOKENS)