Husnain commited on
Commit
bc00a8c
1 Parent(s): 52016f3

💎 [Feature] New model supported: command-r-plus (CohereForai/c4ai-com…

Browse files
Files changed (1) hide show
  1. 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
- "zephyr-orpo-141b": "HuggingFaceH4/zephyr-orpo-141b-A35b-v0.1",
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
- "zephyr-orpo-141b": "<|end_of_turn|>",
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
- "zephyr-orpo-141b": 8192,
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": "zephyr-orpo-141b",
60
- "description": "[HuggingFaceH4/zephyr-orpo-141b]: https://huggingface.co/HuggingFaceH4/zephyr-orpo-141b-A35b-v0.1",
61
  "object": "model",
62
  "created": 1700000000,
63
- "owned_by": "HuggingFaceH4",
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",