-
Notifications
You must be signed in to change notification settings - Fork 0
/
config.toml
69 lines (57 loc) · 1.45 KB
/
config.toml
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
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
examples = [
"Write a poem about AI.",
"Write a Hello World in Rust.",
"Explain algorithmic entropy.",
"Write some example Markdown.",
]
default_model = "NousResearch/Nous-Hermes-2-Mixtral-8x7B-DPO"
[[model]]
model_id = "NousResearch/Nous-Hermes-2-Mixtral-8x7B-DPO"
name = "Nous Hermes 2 - Mixtral 8x7B - DPO"
chat_template = "ChatML"
stream = true
[[model]]
model_id = "mistralai/Mistral-7B-Instruct-v0.2"
name = "Mistral-7B-Instruct-v0.2"
chat_template = "Instruct"
stream = true
[[model]]
model_id = "mistralai/Mistral-7B-v0.1"
name = "Mistral-7B-v0.1"
stream = true
[[model]]
model_id = "mistralai/Mixtral-8x7B-Instruct-v0.1"
name = "Mixtral-8x7B"
chat_template = "Instruct"
stream = true
#[[model]]
#model_id = "codellama/CodeLlama-70b-Instruct-hf"
#chat_template = "CodeLlamaInstruct"
#stream = false
[[model]]
model_id = "TinyLlama/TinyLlama-1.1B-Chat-v1.0"
name = "TinyLlama-1.1B"
chat_template = "ChatML"
stream = false
[[model]]
model_id = "HuggingFaceH4/zephyr-7b-beta"
name = "Zephyr 7B β"
chat_template = "ChatML"
stream = true
[[model]]
model_id = "openai-community/gpt2"
name = "GPT-2"
stream = false
# note: does work (with streaming) but returns prompt
#[[model]]
#model_id = "bigcode/starcoder"
#name = "StarCoder"
#stream = false
[[model]]
model_id = "codellama/CodeLlama-7b-hf"
name = "Code Llama"
stream = true
#[[model]]
#model_id = "cognitivecomputations/TinyDolphin-2.8.1-1.1b"
#name = "TinyDolphin-2.8.1-1.1b"
#stream = false