ehagey commited on
Commit
151ae44
·
verified ·
1 Parent(s): 6ed5a99

Update config.py

Browse files
Files changed (1) hide show
  1. config.py +22 -1
config.py CHANGED
@@ -1,10 +1,31 @@
1
  MODELS = {
2
  "Llama-3-70B-Instruct-Turbo": {
 
3
  "model_id": "meta-llama/Llama-3.3-70B-Instruct-Turbo",
4
  },
5
  "Llama-3-3B-Instruct-Turbo": {
 
6
  "model_id": "meta-llama/Llama-3.2-3B-Instruct-Turbo",
7
-
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
8
  }
9
  }
10
 
 
1
  MODELS = {
2
  "Llama-3-70B-Instruct-Turbo": {
3
+ "provider": "togetherai",
4
  "model_id": "meta-llama/Llama-3.3-70B-Instruct-Turbo",
5
  },
6
  "Llama-3-3B-Instruct-Turbo": {
7
+ "provider": "togetherai",
8
  "model_id": "meta-llama/Llama-3.2-3B-Instruct-Turbo",
9
+ },
10
+ "DeepSeek-R1":{
11
+ "provider": "togetherai",
12
+ "model_id": "deepseek-ai/DeepSeek-R1",
13
+ },
14
+ "GPT-4o mini":{
15
+ "provider": "openai",
16
+ "model_id": "gpt-4o-mini",
17
+ },
18
+ "Claude 3.5 Haiku":{
19
+ "provider": "anthropic",
20
+ "model_id": "claude-3-5-haiku-20241022",
21
+ },
22
+ "Claude 3.5 Sonnet":{
23
+ "provider": "anthropic",
24
+ "model_id": "claude-3-5-sonnet-20241022",
25
+ },
26
+ "GPT 4o":{
27
+ "provider": "openai",
28
+ "model_id": "gpt-4o",
29
  }
30
  }
31