Home-Assistant
/
core
mirror of https://github.com/home-assistant/core.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Change Ollama default to llama3.2 (#129901)

pull/129906/head
Paulus Schoutsen 2024-11-05 11:02:44 -05:00 committed by GitHub
parent 9253fa4471
commit ed56e5d631
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 46 additions and 18 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

64
homeassistant/components/ollama/const.py
View File

@ -24,8 +24,12 @@ MAX_HISTORY_SECONDS = 60 * 60 # 1 hour
MODEL_NAMES = [ # https://ollama.com/library
"alfred",
"all-minilm",
"aya-expanse",
"aya",
"bakllava",
"bespoke-minicheck",
"bge-large",
"bge-m3",
"codebooga",
"codegeex4",
"codegemma",
@ -33,18 +37,19 @@ MODEL_NAMES = [ # https://ollama.com/library
"codeqwen",
"codestral",
"codeup",
"command-r",
"command-r-plus",
"command-r",
"dbrx",
"deepseek-coder",
"deepseek-coder-v2",
"deepseek-coder",
"deepseek-llm",
"deepseek-v2.5",
"deepseek-v2",
"dolphincoder",
"dolphin-llama3",
"dolphin-mistral",
"dolphin-mixtral",
"dolphin-phi",
"dolphincoder",
"duckdb-nsql",
"everythinglm",
"falcon",
@ -55,74 +60,97 @@ MODEL_NAMES = [ # https://ollama.com/library
"glm4",
"goliath",
"granite-code",
"granite3-dense",
"granite3-guardian" "granite3-moe",
"hermes3",
"internlm2",
"llama2",
"llama-guard3",
"llama-pro",
"llama2-chinese",
"llama2-uncensored",
"llama3",
"llama2",
"llama3-chatqa",
"llama3-gradient",
"llama3-groq-tool-use",
"llama-pro",
"llava",
"llama3.1",
"llama3.2",
"llama3",
"llava-llama3",
"llava-phi3",
"llava",
"magicoder",
"mathstral",
"meditron",
"medllama2",
"megadolphin",
"mistral",
"mistrallite",
"minicpm-v",
"mistral-large",
"mistral-nemo",
"mistral-openorca",
"mistral-small",
"mistral",
"mistrallite",
"mixtral",
"moondream",
"mxbai-embed-large",
"nemotron-mini",
"nemotron",
"neural-chat",
"nexusraven",
"nomic-embed-text",
"notus",
"notux",
"nous-hermes",
"nous-hermes2",
"nous-hermes2-mixtral",
"nous-hermes2",
"nuextract",
"open-orca-platypus2",
"openchat",
"openhermes",
"open-orca-platypus2",
"orca2",
"orca-mini",
"orca2",
"paraphrase-multilingual",
"phi",
"phi3.5",
"phi3",
"phind-codellama",
"qwen",
"qwen2-math",
"qwen2.5-coder",
"qwen2.5",
"qwen2",
"reader-lm",
"reflection",
"samantha-mistral",
"shieldgemma",
"smollm",
"smollm2",
"snowflake-arctic-embed",
"solar-pro",
"solar",
"sqlcoder",
"stable-beluga",
"stable-code",
"stablelm2",
"stablelm-zephyr",
"stablelm2",
"starcoder",
"starcoder2",
"starling-lm",
"tinydolphin",
"tinyllama",
"vicuna",
"wizard-math",
"wizard-vicuna-uncensored",
"wizard-vicuna",
"wizardcoder",
"wizardlm-uncensored",
"wizardlm",
"wizardlm2",
"wizardlm-uncensored",
"wizard-math",
"wizard-vicuna",
"wizard-vicuna-uncensored",
"xwinlm",
"yarn-llama2",
"yarn-mistral",
"yi-coder",
"yi",
"zephyr",
]
DEFAULT_MODEL = "llama3.1:latest"
DEFAULT_MODEL = "llama3.2:latest"