Husnain
commited on
💎 [Feature] New model supported: command-r-plus (CohereForai/c4ai-com…
Browse files- constants/models.py +6 -6
constants/models.py
CHANGED
|
@@ -2,7 +2,7 @@ MODEL_MAP = {
|
|
| 2 |
"mixtral-8x7b": "mistralai/Mixtral-8x7B-Instruct-v0.1", # [Recommended]
|
| 3 |
"nous-mixtral-8x7b": "NousResearch/Nous-Hermes-2-Mixtral-8x7B-DPO",
|
| 4 |
"mistral-7b": "mistralai/Mistral-7B-Instruct-v0.2",
|
| 5 |
-
"
|
| 6 |
"gemma-1.1-7b": "google/gemma-1.1-7b-it",
|
| 7 |
"command-r-plus": "CohereForAI/c4ai-command-r-plus",
|
| 8 |
"default": "mistralai/Mixtral-8x7B-Instruct-v0.1",
|
|
@@ -14,7 +14,7 @@ STOP_SEQUENCES_MAP = {
|
|
| 14 |
"mixtral-8x7b": "</s>",
|
| 15 |
"nous-mixtral-8x7b": "<|im_end|>",
|
| 16 |
"mistral-7b": "</s>",
|
| 17 |
-
"
|
| 18 |
"gemma-1.1-7b": "<eos>",
|
| 19 |
"command-r-plus": "<|END_OF_TURN_TOKEN|>",
|
| 20 |
}
|
|
@@ -23,7 +23,7 @@ TOKEN_LIMIT_MAP = {
|
|
| 23 |
"mixtral-8x7b": 32768,
|
| 24 |
"nous-mixtral-8x7b": 32768,
|
| 25 |
"mistral-7b": 32768,
|
| 26 |
-
"
|
| 27 |
"gemma-1.1-7b": 8192,
|
| 28 |
"gpt-3.5-turbo": 8192,
|
| 29 |
"command-r-plus": 32768,
|
|
@@ -56,11 +56,11 @@ AVAILABLE_MODELS_DICTS = [
|
|
| 56 |
"owned_by": "mistralai",
|
| 57 |
},
|
| 58 |
{
|
| 59 |
-
"id": "
|
| 60 |
-
"description": "[
|
| 61 |
"object": "model",
|
| 62 |
"created": 1700000000,
|
| 63 |
-
"owned_by": "
|
| 64 |
},
|
| 65 |
{
|
| 66 |
"id": "gemma-1.1-7b",
|
|
|
|
| 2 |
"mixtral-8x7b": "mistralai/Mixtral-8x7B-Instruct-v0.1", # [Recommended]
|
| 3 |
"nous-mixtral-8x7b": "NousResearch/Nous-Hermes-2-Mixtral-8x7B-DPO",
|
| 4 |
"mistral-7b": "mistralai/Mistral-7B-Instruct-v0.2",
|
| 5 |
+
"openchat-3.5": "openchat/openchat-3.5-0106",
|
| 6 |
"gemma-1.1-7b": "google/gemma-1.1-7b-it",
|
| 7 |
"command-r-plus": "CohereForAI/c4ai-command-r-plus",
|
| 8 |
"default": "mistralai/Mixtral-8x7B-Instruct-v0.1",
|
|
|
|
| 14 |
"mixtral-8x7b": "</s>",
|
| 15 |
"nous-mixtral-8x7b": "<|im_end|>",
|
| 16 |
"mistral-7b": "</s>",
|
| 17 |
+
"openchat-3.5": "<|end_of_turn|>",
|
| 18 |
"gemma-1.1-7b": "<eos>",
|
| 19 |
"command-r-plus": "<|END_OF_TURN_TOKEN|>",
|
| 20 |
}
|
|
|
|
| 23 |
"mixtral-8x7b": 32768,
|
| 24 |
"nous-mixtral-8x7b": 32768,
|
| 25 |
"mistral-7b": 32768,
|
| 26 |
+
"openchat-3.5": 8192,
|
| 27 |
"gemma-1.1-7b": 8192,
|
| 28 |
"gpt-3.5-turbo": 8192,
|
| 29 |
"command-r-plus": 32768,
|
|
|
|
| 56 |
"owned_by": "mistralai",
|
| 57 |
},
|
| 58 |
{
|
| 59 |
+
"id": "openchat-3.5",
|
| 60 |
+
"description": "[openchat/openchat-3.5-0106]: https://huggingface.co/openchat/openchat-3.5-0106",
|
| 61 |
"object": "model",
|
| 62 |
"created": 1700000000,
|
| 63 |
+
"owned_by": "openchat",
|
| 64 |
},
|
| 65 |
{
|
| 66 |
"id": "gemma-1.1-7b",
|