forked from danielgross/localpilot
-
Notifications
You must be signed in to change notification settings - Fork 0
/
config.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')