Spaces:
Paused
Paused
: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",
|