Husnain commited on
Commit
14a9a52
1 Parent(s): dcce9b7

♻️ [Refactor] change gemma-7b to gemma-7b-1.1

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-1.1-7b-it",
7
  "default": "mistralai/Mixtral-8x7B-Instruct-v0.1",
8
  }
9
 
@@ -13,7 +13,7 @@ STOP_SEQUENCES_MAP = {
13
  "nous-mixtral-8x7b": "<|im_end|>",
14
  "mistral-7b": "</s>",
15
  "openchat-3.5": "<|end_of_turn|>",
16
- "gemma-7b": "<eos>",
17
  }
18
 
19
  TOKEN_LIMIT_MAP = {
@@ -21,7 +21,7 @@ TOKEN_LIMIT_MAP = {
21
  "nous-mixtral-8x7b": 32768,
22
  "mistral-7b": 32768,
23
  "openchat-3.5": 8192,
24
- "gemma-7b": 8192,
25
  "gpt-3.5-turbo": 8192,
26
  }
27
 
@@ -33,7 +33,7 @@ AVAILABLE_MODELS = [
33
  "nous-mixtral-8x7b",
34
  "mistral-7b",
35
  "openchat-3.5",
36
- "gemma-7b",
37
  "gpt-3.5-turbo",
38
  ]
39
 
@@ -68,7 +68,7 @@ AVAILABLE_MODELS_DICTS = [
68
  "owned_by": "openchat",
69
  },
70
  {
71
- "id": "gemma-7b",
72
  "description": "[google/gemma-1.1-7b-it]: https://huggingface.co/google/gemma-1.1-7b-it",
73
  "object": "model",
74
  "created": 1700000000,
 
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-1.1": "google/gemma-1.1-7b-it",
7
  "default": "mistralai/Mixtral-8x7B-Instruct-v0.1",
8
  }
9
 
 
13
  "nous-mixtral-8x7b": "<|im_end|>",
14
  "mistral-7b": "</s>",
15
  "openchat-3.5": "<|end_of_turn|>",
16
+ "gemma-7b-1.1": "<eos>",
17
  }
18
 
19
  TOKEN_LIMIT_MAP = {
 
21
  "nous-mixtral-8x7b": 32768,
22
  "mistral-7b": 32768,
23
  "openchat-3.5": 8192,
24
+ "gemma-7b-1.1": 8192,
25
  "gpt-3.5-turbo": 8192,
26
  }
27
 
 
33
  "nous-mixtral-8x7b",
34
  "mistral-7b",
35
  "openchat-3.5",
36
+ "-7b-1.1",
37
  "gpt-3.5-turbo",
38
  ]
39
 
 
68
  "owned_by": "openchat",
69
  },
70
  {
71
+ "id": "gemma-7b-1.1",
72
  "description": "[google/gemma-1.1-7b-it]: https://huggingface.co/google/gemma-1.1-7b-it",
73
  "object": "model",
74
  "created": 1700000000,