Spaces:
Running
Running
Husnain
commited on
Commit
•
bc00a8c
1
Parent(s):
52016f3
💎 [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",
|