format fix

zed-industries · Nov 20, 2024 · 39fdb1e · 39fdb1e
1 parent 804db35
commit 39fdb1e
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/crates/ollama/src/ollama.rs b/crates/ollama/src/ollama.rs
@@ -81,7 +81,8 @@ fn get_max_tokens(name: &str) -> usize {
         "llama2" | "yi" | "vicuna" | "stablelm2" => 4096,
         "llama3" | "gemma2" | "gemma" | "codegemma" | "starcoder" | "aya" => 8192,
         "codellama" | "starcoder2" => 16384,
-        "mistral" | "codestral" | "mixstral" | "llava" | "qwen2" | "qwen2.5-coder" | "dolphin-mixtral" => 32768,
+        "mistral" | "codestral" | "mixstral" | "llava" | "qwen2" | "qwen2.5-coder"
+        | "dolphin-mixtral" => 32768,
         "llama3.1" | "phi3" | "phi3.5" | "command-r" | "deepseek-coder-v2" | "yi-coder"
         | "llama3.2" => 128000,
         _ => DEFAULT_TOKENS,