Spaces:
Running
Running
Husnain
commited on
Commit
•
14a9a52
1
Parent(s):
dcce9b7
♻️ [Refactor] change gemma-7b to gemma-7b-1.1
Browse files- 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 |
-
"
|
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,
|