Hopiu/llm
1
0
Fork 0
mirror of https://github.com/Hopiu/llm.git synced 2026-04-08 07:20:58 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Add 16k models (#38)

Browse source 
also added a `3.5` alias
This commit is contained in:
Benjamin Kirkbride 2023-06-17 04:22:52 -04:00 committed by GitHub
parent 5e21195e93
commit b12e28c4ec
1 changed files with 8 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
llm/cli.py
View file

@ -20,7 +20,14 @@ warnings.simplefilter("ignore", ResourceWarning)
DEFAULT_MODEL = "gpt-3.5-turbo"
MODEL_ALIASES = {"4": "gpt-4", "gpt4": "gpt-4", "chatgpt": "gpt-3.5-turbo"}
MODEL_ALIASES = {
"4": "gpt-4",
"gpt4": "gpt-4",
"chatgpt": "gpt-3.5-turbo",
"3.5": "gpt-3.5-turbo",
"chatgpt-16k": "gpt-3.5-turbo-16k",
"3.5-16k": "gpt-3.5-turbo-16k",
}
DEFAULT_TEMPLATE = "prompt: "