Husnain commited on
Commit
c18033f
1 Parent(s): e9595c5

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

Browse files
Files changed (1) hide show
  1. constants/models.py +5 -5
constants/models.py CHANGED
@@ -3,7 +3,7 @@ MODEL_MAP = {
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-7b": "google/gemma-7b-it",
7
  "command-r-plus": "CohereForAI/c4ai-command-r-plus",
8
  "default": "mistralai/Mixtral-8x7B-Instruct-v0.1",
9
  }
@@ -15,7 +15,7 @@ STOP_SEQUENCES_MAP = {
15
  "nous-mixtral-8x7b": "<|im_end|>",
16
  "mistral-7b": "</s>",
17
  "openchat-3.5": "<|end_of_turn|>",
18
- "gemma-7b": "<eos>",
19
  "command-r-plus": "<|END_OF_TURN_TOKEN|>",
20
  }
21
 
@@ -24,7 +24,7 @@ TOKEN_LIMIT_MAP = {
24
  "nous-mixtral-8x7b": 32768,
25
  "mistral-7b": 32768,
26
  "openchat-3.5": 8192,
27
- "gemma-7b": 8192,
28
  "gpt-3.5-turbo": 8192,
29
  "command-r-plus": 32768,
30
  }
@@ -63,8 +63,8 @@ AVAILABLE_MODELS_DICTS = [
63
  "owned_by": "openchat",
64
  },
65
  {
66
- "id": "gemma-7b",
67
- "description": "[google/gemma-7b-it]: https://huggingface.co/google/gemma-7b-it",
68
  "object": "model",
69
  "created": 1700000000,
70
  "owned_by": "Google",
 
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",
9
  }
 
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
  }
21
 
 
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,
30
  }
 
63
  "owned_by": "openchat",
64
  },
65
  {
66
+ "id": "gemma-1.1-7b-it",
67
+ "description": "[google/gemma-1.1-7b-it]: https://huggingface.co/google/gemma-1.1-7b-it",
68
  "object": "model",
69
  "created": 1700000000,
70
  "owned_by": "Google",