diff --git a/nvim/lua/plugins/nvim-cmp.lua b/nvim/lua/plugins/nvim-cmp.lua index 2f58af9..1fa5694 100644 --- a/nvim/lua/plugins/nvim-cmp.lua +++ b/nvim/lua/plugins/nvim-cmp.lua @@ -45,9 +45,7 @@ return { elseif model:match '^qwen' then return { '<|endoftext|>' } elseif model:match '^starcoder' then - return { '' } - elseif model:match '^codegemma' then - return { '<|fim_prefix|>', '<|fim_suffix|>', '<|fim_middle|>', '<|file_separator|>' } + return { '', '<|end_of_text|>', '<|endoftext|>' } elseif model:match '^codestral' then return { '[INST]', '[/INST]', '[PREFIX]', '[MIDDLE]', '[SUFFIX]' } elseif model:match '^deepseek-coder' then @@ -57,12 +55,11 @@ return { local llama_base_url = os.getenv 'LLAMA_API_BASE' or 'http://localhost:11434' -- local model = 'codellama:7b-code' - -- local model = 'codellama:13b-code' - -- local model = 'qwen2.5-coder:7b-base' + local model = 'codellama:13b-code' + -- local model = 'qwen2.5-coder:7b' -- local model = 'starcoder2:7b' -- local model = 'codestral:22b' - -- local model = 'codegemma:7b-code' - local model = 'deepseek-coder-v2:16b' + -- local model = 'deepseek-coder-v2:16b' require('minuet').setup { enabled = true,