evlli/hass-core
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Update Ollama model names list (#116172)

Browse source
This commit is contained in:
Michael Hansen 2024-04-25 13:35:29 -05:00 • committed by Paulus Schoutsen
parent 12bce5451e
commit 5ac8488d2a
1 changed files with 78 additions and 67 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

145
homeassistant/components/ollama/const.py
View file

@ -81,75 +81,86 @@ DEFAULT_MAX_HISTORY = 20
MAX_HISTORY_SECONDS = 60 * 60 # 1 hour MAX_HISTORY_SECONDS = 60 * 60 # 1 hour
MODEL_NAMES = [ # https://ollama.com/library MODEL_NAMES = [ # https://ollama.com/library
"gemma",
"llama2",
"mistral",
"mixtral",
"llava",
"neural-chat",
"codellama",
"dolphin-mixtral",
"qwen",
"llama2-uncensored",
"mistral-openorca",
"deepseek-coder",
"nous-hermes2",
"phi",
"orca-mini",
"dolphin-mistral",
"wizard-vicuna-uncensored",
"vicuna",
"tinydolphin",
"llama2-chinese",
"nomic-embed-text",
"openhermes",
"zephyr",
"tinyllama",
"openchat",
"wizardcoder",
"starcoder",
"phind-codellama",
"starcoder2",
"yi",
"orca2",
"falcon",
"wizard-math",
"dolphin-phi",
"starling-lm",
"nous-hermes",
"stable-code",
"medllama2",
"bakllava",
"codeup",
"wizardlm-uncensored",
"solar",
"everythinglm",
"sqlcoder",
"dolphincoder",
"nous-hermes2-mixtral",
"stable-beluga",
"yarn-mistral",
"stablelm2",
"samantha-mistral",
"meditron",
"stablelm-zephyr",
"magicoder",
"yarn-llama2",
"llama-pro",
"deepseek-llm",
"wizard-vicuna",
"codebooga",
"mistrallite",
"all-minilm",
"nexusraven",
"open-orca-platypus2",
"goliath",
"notux",
"megadolphin",
"alfred", "alfred",
"xwinlm", "all-minilm",
"wizardlm", "bakllava",
"codebooga",
"codegemma",
"codellama",
"codeqwen",
"codeup",
"command-r",
"command-r-plus",
"dbrx",
"deepseek-coder",
"deepseek-llm",
"dolphin-llama3",
"dolphin-mistral",
"dolphin-mixtral",
"dolphin-phi",
"dolphincoder",
"duckdb-nsql", "duckdb-nsql",
"everythinglm",
"falcon",
"gemma",
"goliath",
"llama-pro",
"llama2",
"llama2-chinese",
"llama2-uncensored",
"llama3",
"llava",
"magicoder",
"meditron",
"medllama2",
"megadolphin",
"mistral",
"mistral-openorca",
"mistrallite",
"mixtral",
"mxbai-embed-large",
"neural-chat",
"nexusraven",
"nomic-embed-text",
"notus", "notus",
"notux",
"nous-hermes",
"nous-hermes2",
"nous-hermes2-mixtral",
"open-orca-platypus2",
"openchat",
"openhermes",
"orca-mini",
"orca2",
"phi",
"phi3",
"phind-codellama",
"qwen",
"samantha-mistral",
"snowflake-arctic-embed",
"solar",
"sqlcoder",
"stable-beluga",
"stable-code",
"stablelm-zephyr",
"stablelm2",
"starcoder",
"starcoder2",
"starling-lm",
"tinydolphin",
"tinyllama",
"vicuna",
"wizard-math",
"wizard-vicuna",
"wizard-vicuna-uncensored",
"wizardcoder",
"wizardlm",
"wizardlm-uncensored",
"wizardlm2",
"xwinlm",
"yarn-llama2",
"yarn-mistral",
"yi",
"zephyr",
] ]
DEFAULT_MODEL = "llama2:latest" DEFAULT_MODEL = "llama2:latest"