forked from danielgross/localpilot
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathconfig.py
26 lines (24 loc) · 899 Bytes
/
config.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
import os
models = {
"GitHub": {
"domain": "https://copilot-proxy.githubusercontent.com",
"type": "remote",
},
"CodeLlama-7b": {
"url": "https://huggingface.co/TheBloke/CodeLlama-7B-GGUF/resolve/main/codellama-7b.Q5_K_S.gguf",
"type": "local",
"filename": "codellama-7b.Q5_K_S.gguf",
},
"Mistral-7b": {
"url": "https://huggingface.co/TheBloke/Mistral-7B-Instruct-v0.1-GGUF/resolve/main/mistral-7b-instruct-v0.1.Q5_K_M.gguf",
"type": "local",
"filename": "mistral-7b-instruct-v0.1.Q5_K_M.gguf",
},
"CodeLlama-34b": {
"url": "https://huggingface.co/TheBloke/CodeLlama-34B-Instruct-GGUF/resolve/main/codellama-34b-instruct.Q4_K_M.gguf",
"type": "local",
"filename": "codellama-34b-instruct.Q4_K_M.gguf",
},
"default": "GitHub",
}
model_folder = os.path.expanduser("~/models")