Kang Suhyun commited on
Commit
9f68c4f
1 Parent(s): fd9a72d

[#128] Add Llama3.1 models and update GPT models (#133)

Browse files
Files changed (2) hide show
  1. README.md +0 -1
  2. model.py +10 -34
README.md CHANGED
@@ -54,7 +54,6 @@ Get Involved: [Discuss and contribute on GitHub](https://github.com/yanolja/aren
54
  TRANSLATIONS_COLLECTION=<your collection> \
55
  OPENAI_API_KEY=<your key> \
56
  ANTHROPIC_API_KEY=<your key> \
57
- MISTRAL_API_KEY=<your key> \
58
  GEMINI_API_KEY=<your key> \
59
  DEEPINFRA_API_KEY=<your key> \
60
  python3 app.py
 
54
  TRANSLATIONS_COLLECTION=<your collection> \
55
  OPENAI_API_KEY=<your key> \
56
  ANTHROPIC_API_KEY=<your key> \
 
57
  GEMINI_API_KEY=<your key> \
58
  DEEPINFRA_API_KEY=<your key> \
59
  python3 app.py
model.py CHANGED
@@ -142,45 +142,21 @@ class VertexModel(Model):
142
  }
143
 
144
 
145
- class EeveModel(Model):
146
-
147
- def _get_completion_kwargs(self):
148
- json_template = {
149
- "type": "object",
150
- "properties": {
151
- "result": {
152
- "type": "string"
153
- }
154
- }
155
- }
156
- return {
157
- "extra_body": {
158
- "guided_json": json.dumps(json_template),
159
- "guided_decoding_backend": "lm-format-enforcer"
160
- }
161
- }
162
-
163
-
164
  supported_models: List[Model] = [
165
- Model("gpt-4o-2024-05-13"),
166
- Model("gpt-4-turbo-2024-04-09"),
167
- Model("gpt-4-0125-preview"),
168
- Model("gpt-3.5-turbo-0125"),
169
- AnthropicModel("claude-3-opus-20240229"),
170
- AnthropicModel("claude-3-sonnet-20240229"),
171
- AnthropicModel("claude-3-haiku-20240307"),
172
  VertexModel("gemini-1.5-pro-001",
173
  vertex_credentials=os.getenv("VERTEX_CREDENTIALS")),
174
- Model("mistral-small-2402", provider="mistral"),
175
- Model("mistral-large-2402", provider="mistral"),
176
- Model("meta-llama/Meta-Llama-3-8B-Instruct", provider="deepinfra"),
177
- Model("meta-llama/Meta-Llama-3-70B-Instruct", provider="deepinfra"),
 
 
 
178
  Model("google/gemma-2-9b-it", provider="deepinfra"),
179
  Model("google/gemma-2-27b-it", provider="deepinfra"),
180
- EeveModel("yanolja/EEVE-Korean-Instruct-10.8B-v1.0",
181
- provider="openai",
182
- api_base=os.getenv("EEVE_API_BASE"),
183
- api_key=os.getenv("EEVE_API_KEY")),
184
  ]
185
 
186
 
 
142
  }
143
 
144
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
145
  supported_models: List[Model] = [
146
+ Model("gpt-4o-2024-08-06"),
147
+ Model("gpt-4o-mini-2024-07-18"),
148
+ AnthropicModel("claude-3-5-sonnet-20240620"),
 
 
 
 
149
  VertexModel("gemini-1.5-pro-001",
150
  vertex_credentials=os.getenv("VERTEX_CREDENTIALS")),
151
+ VertexModel("gemini-1.5-flash-preview-0514",
152
+ vertex_credentials=os.getenv("VERTEX_CREDENTIALS")),
153
+ Model("meta-llama/Meta-Llama-3.1-8B-Instruct", provider="deepinfra"),
154
+ Model("meta-llama/Meta-Llama-3.1-70B-Instruct", provider="deepinfra"),
155
+ Model("meta-llama/Meta-Llama-3.1-405B-Instruct", provider="deepinfra"),
156
+ Model("Qwen/Qwen2.5-72B-Instruct", provider="deepinfra"),
157
+ Model("Qwen/Qwen2-72B-Instruct", provider="deepinfra"),
158
  Model("google/gemma-2-9b-it", provider="deepinfra"),
159
  Model("google/gemma-2-27b-it", provider="deepinfra"),
 
 
 
 
160
  ]
161
 
162