Spaces:
Running
Running
Kang Suhyun
commited on
Commit
•
9f68c4f
1
Parent(s):
fd9a72d
[#128] Add Llama3.1 models and update GPT models (#133)
Browse files
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-
|
166 |
-
Model("gpt-
|
167 |
-
|
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 |
-
|
175 |
-
|
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 |
|