返回提交历史
Deleted
g4f/models.py
+0
-218
XFEstudio/gpt4free
.
f335062d
代码差异
1 个文件
+0
-218
@@ -1,218 +0,0 @@
1
from __future__ import annotations
2
3
from dataclasses import dataclass
4
5
from .Provider import (Aichat, Aivvm, ChatBase, ChatgptAi, ChatgptLogin, CodeLinkAva, BaseProvider, Bard, H2o, DeepAi,
6
Vercel, Vitalentum, Ylokh, You, Yqcloud)
7
from .typing import Union
8
9
10
@dataclass
11
class Model:
12
name: str
13
base_provider: str
14
best_provider: Union[type[BaseProvider], list[type[BaseProvider]]]
15
16
17
# Config for HuggingChat, OpenAssistant
18
# Works for Liaobots, H2o, OpenaiChat, Yqcloud, You
19
default = Model(
20
name="",
21
base_provider="huggingface",
22
best_provider=H2o,
23
)
24
25
# GPT-3.5 / GPT-4
26
gpt_35_turbo = Model(
27
name='gpt-3.5-turbo',
28
base_provider='openai',
29
best_provider=[
30
Vercel, Aichat, Aivvm, ChatBase, ChatgptAi, ChatgptLogin,
31
CodeLinkAva, DeepAi, Vitalentum, Ylokh, You, Yqcloud
32
]
33
)
34
35
gpt_4 = Model(
36
name='gpt-4',
37
base_provider='openai',
38
best_provider=Aivvm
39
)
40
41
# Bard
42
palm = Model(
43
name='palm',
44
base_provider='google',
45
best_provider=Bard)
46
47
# H2o
48
falcon_7b = Model(
49
name='h2oai/h2ogpt-gm-oasst1-en-2048-falcon-7b-v3',
50
base_provider='huggingface',
51
best_provider=H2o)
52
53
falcon_40b = Model(
54
name='h2oai/h2ogpt-gm-oasst1-en-2048-falcon-40b-v1',
55
base_provider='huggingface',
56
best_provider=H2o)
57
58
llama_13b = Model(
59
name='h2oai/h2ogpt-gm-oasst1-en-2048-open-llama-13b',
60
base_provider='huggingface',
61
best_provider=H2o)
62
63
# Vercel
64
claude_instant_v1 = Model(
65
name='anthropic:claude-instant-v1',
66
base_provider='anthropic',
67
best_provider=Vercel)
68
69
claude_v1 = Model(
70
name='anthropic:claude-v1',
71
base_provider='anthropic',
72
best_provider=Vercel)
73
74
claude_v2 = Model(
75
name='anthropic:claude-v2',
76
base_provider='anthropic',
77
best_provider=Vercel)
78
79
command_light_nightly = Model(
80
name='cohere:command-light-nightly',
81
base_provider='cohere',
82
best_provider=Vercel)
83
84
command_nightly = Model(
85
name='cohere:command-nightly',
86
base_provider='cohere',
87
best_provider=Vercel)
88
89
gpt_neox_20b = Model(
90
name='huggingface:EleutherAI/gpt-neox-20b',
91
base_provider='huggingface',
92
best_provider=Vercel)
93
94
oasst_sft_1_pythia_12b = Model(
95
name='huggingface:OpenAssistant/oasst-sft-1-pythia-12b',
96
base_provider='huggingface',
97
best_provider=Vercel)
98
99
oasst_sft_4_pythia_12b_epoch_35 = Model(
100
name='huggingface:OpenAssistant/oasst-sft-4-pythia-12b-epoch-3.5',
101
base_provider='huggingface',
102
best_provider=Vercel)
103
104
santacoder = Model(
105
name='huggingface:bigcode/santacoder',
106
base_provider='huggingface',
107
best_provider=Vercel)
108
109
bloom = Model(
110
name='huggingface:bigscience/bloom',
111
base_provider='huggingface',
112
best_provider=Vercel)
113
114
flan_t5_xxl = Model(
115
name='huggingface:google/flan-t5-xxl',
116
base_provider='huggingface',
117
best_provider=Vercel)
118
119
code_davinci_002 = Model(
120
name='openai:code-davinci-002',
121
base_provider='openai',
122
best_provider=Vercel)
123
124
gpt_35_turbo_16k = Model(
125
name='openai:gpt-3.5-turbo-16k',
126
base_provider='openai',
127
best_provider=Vercel)
128
129
gpt_35_turbo_16k_0613 = Model(
130
name='openai:gpt-3.5-turbo-16k-0613',
131
base_provider='openai',
132
best_provider=Vercel)
133
134
gpt_4_0613 = Model(
135
name='openai:gpt-4-0613',
136
base_provider='openai',
137
best_provider=Vercel)
138
139
text_ada_001 = Model(
140
name='openai:text-ada-001',
141
base_provider='openai',
142
best_provider=Vercel)
143
144
text_babbage_001 = Model(
145
name='openai:text-babbage-001',
146
base_provider='openai',
147
best_provider=Vercel)
148
149
text_curie_001 = Model(
150
name='openai:text-curie-001',
151
base_provider='openai',
152
best_provider=Vercel)
153
154
text_davinci_002 = Model(
155
name='openai:text-davinci-002',
156
base_provider='openai',
157
best_provider=Vercel)
158
159
text_davinci_003 = Model(
160
name='openai:text-davinci-003',
161
base_provider='openai',
162
best_provider=Vercel)
163
164
llama13b_v2_chat = Model(
165
name='replicate:a16z-infra/llama13b-v2-chat',
166
base_provider='replicate',
167
best_provider=Vercel)
168
169
llama7b_v2_chat = Model(
170
name='replicate:a16z-infra/llama7b-v2-chat',
171
base_provider='replicate',
172
best_provider=Vercel)
173
174
175
class ModelUtils:
176
convert: dict[str, Model] = {
177
# GPT-3.5 / GPT-4
178
'gpt-3.5-turbo': gpt_35_turbo,
179
'gpt-4': gpt_4,
180
181
# Bard
182
'palm2': palm,
183
'palm': palm,
184
'google': palm,
185
'google-bard': palm,
186
'google-palm': palm,
187
'bard': palm,
188
189
# H2o
190
'falcon-40b': falcon_40b,
191
'falcon-7b': falcon_7b,
192
'llama-13b': llama_13b,
193
194
# Vercel
195
'claude-instant-v1': claude_instant_v1,
196
'claude-v1': claude_v1,
197
'claude-v2': claude_v2,
198
'command-nightly': command_nightly,
199
'gpt-neox-20b': gpt_neox_20b,
200
'santacoder': santacoder,
201
'bloom': bloom,
202
'flan-t5-xxl': flan_t5_xxl,
203
'code-davinci-002': code_davinci_002,
204
'gpt-3.5-turbo-16k': gpt_35_turbo_16k,
205
'gpt-4-0613': gpt_4_0613,
206
'text-ada-001': text_ada_001,
207
'text-babbage-001': text_babbage_001,
208
'text-curie-001': text_curie_001,
209
'text-davinci-002': text_davinci_002,
210
'text-davinci-003': text_davinci_003,
211
'llama13b-v2-chat': llama13b_v2_chat,
212
'llama7b-v2-chat': llama7b_v2_chat,
213
214
'oasst-sft-1-pythia-12b': oasst_sft_1_pythia_12b,
215
'oasst-sft-4-pythia-12b-epoch-3.5': oasst_sft_4_pythia_12b_epoch_35,
216
'command-light-nightly': command_light_nightly,
217
'gpt-3.5-turbo-16k-0613': gpt_35_turbo_16k_0613,
218
}