:gem: [Feature] New available models and related token limit
Browse files- constants/models.py +24 -8
constants/models.py
CHANGED
@@ -6,6 +6,7 @@ MODEL_MAP = {
|
|
6 |
"gemma-7b": "google/gemma-1.1-7b-it",
|
7 |
"command-r-plus": "CohereForAI/c4ai-command-r-plus",
|
8 |
"llama3-70b": "meta-llama/Meta-Llama-3-70B-Instruct",
|
|
|
9 |
"default": "mistralai/Mixtral-8x7B-Instruct-v0.1",
|
10 |
}
|
11 |
|
@@ -26,9 +27,10 @@ TOKEN_LIMIT_MAP = {
|
|
26 |
"mistral-7b": 32768,
|
27 |
"openchat-3.5": 8192,
|
28 |
"gemma-7b": 8192,
|
29 |
-
"gpt-3.5-turbo": 8192,
|
30 |
"command-r-plus": 32768,
|
31 |
"llama3-70b": 8192,
|
|
|
|
|
32 |
}
|
33 |
|
34 |
TOKEN_RESERVED = 20
|
@@ -71,13 +73,27 @@ AVAILABLE_MODELS_DICTS = [
|
|
71 |
"created": 1700000000,
|
72 |
"owned_by": "Google",
|
73 |
},
|
74 |
-
|
75 |
-
|
76 |
-
|
77 |
-
|
78 |
-
|
79 |
-
|
80 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
81 |
{
|
82 |
"id": "gpt-3.5-turbo",
|
83 |
"description": "[openai/gpt-3.5-turbo]: https://platform.openai.com/docs/models/gpt-3-5-turbo",
|
|
|
6 |
"gemma-7b": "google/gemma-1.1-7b-it",
|
7 |
"command-r-plus": "CohereForAI/c4ai-command-r-plus",
|
8 |
"llama3-70b": "meta-llama/Meta-Llama-3-70B-Instruct",
|
9 |
+
"zephyr-141b": "HuggingFaceH4/zephyr-orpo-141b-A35b-v0.1",
|
10 |
"default": "mistralai/Mixtral-8x7B-Instruct-v0.1",
|
11 |
}
|
12 |
|
|
|
27 |
"mistral-7b": 32768,
|
28 |
"openchat-3.5": 8192,
|
29 |
"gemma-7b": 8192,
|
|
|
30 |
"command-r-plus": 32768,
|
31 |
"llama3-70b": 8192,
|
32 |
+
"zephyr-141b": 2048,
|
33 |
+
"gpt-3.5-turbo": 8192,
|
34 |
}
|
35 |
|
36 |
TOKEN_RESERVED = 20
|
|
|
73 |
"created": 1700000000,
|
74 |
"owned_by": "Google",
|
75 |
},
|
76 |
+
{
|
77 |
+
"id": "command-r-plus",
|
78 |
+
"description": "[CohereForAI/c4ai-command-r-plus]: https://huggingface.co/CohereForAI/c4ai-command-r-plus",
|
79 |
+
"object": "model",
|
80 |
+
"created": 1700000000,
|
81 |
+
"owned_by": "CohereForAI",
|
82 |
+
},
|
83 |
+
{
|
84 |
+
"id": "llama3-70b",
|
85 |
+
"description": "[meta-llama/Meta-Llama-3-70B]: https://huggingface.co/meta-llama/Meta-Llama-3-70B",
|
86 |
+
"object": "model",
|
87 |
+
"created": 1700000000,
|
88 |
+
"owned_by": "Meta",
|
89 |
+
},
|
90 |
+
{
|
91 |
+
"id": "zephyr-141b",
|
92 |
+
"description": "[HuggingFaceH4/zephyr-orpo-141b-A35b-v0.1]: https://huggingface.co/HuggingFaceH4/zephyr-orpo-141b-A35b-v0.1",
|
93 |
+
"object": "model",
|
94 |
+
"created": 1700000000,
|
95 |
+
"owned_by": "Huggingface",
|
96 |
+
},
|
97 |
{
|
98 |
"id": "gpt-3.5-turbo",
|
99 |
"description": "[openai/gpt-3.5-turbo]: https://platform.openai.com/docs/models/gpt-3-5-turbo",
|