Hansimov commited on
Commit
bcf3537
1 Parent(s): c43287d

:gem: [Feature] New available models and related token limit

Browse files
Files changed (1) hide show
  1. constants/models.py +24 -8
constants/models.py CHANGED
@@ -6,6 +6,7 @@ MODEL_MAP = {
6
  "gemma-7b": "google/gemma-1.1-7b-it",
7
  "command-r-plus": "CohereForAI/c4ai-command-r-plus",
8
  "llama3-70b": "meta-llama/Meta-Llama-3-70B-Instruct",
 
9
  "default": "mistralai/Mixtral-8x7B-Instruct-v0.1",
10
  }
11
 
@@ -26,9 +27,10 @@ TOKEN_LIMIT_MAP = {
26
  "mistral-7b": 32768,
27
  "openchat-3.5": 8192,
28
  "gemma-7b": 8192,
29
- "gpt-3.5-turbo": 8192,
30
  "command-r-plus": 32768,
31
  "llama3-70b": 8192,
 
 
32
  }
33
 
34
  TOKEN_RESERVED = 20
@@ -71,13 +73,27 @@ AVAILABLE_MODELS_DICTS = [
71
  "created": 1700000000,
72
  "owned_by": "Google",
73
  },
74
- # {
75
- # "id": "command-r-plus",
76
- # "description": "[CohereForAI/c4ai-command-r-plus]: https://huggingface.co/CohereForAI/c4ai-command-r-plus",
77
- # "object": "model",
78
- # "created": 1700000000,
79
- # "owned_by": "CohereForAI",
80
- # },
 
 
 
 
 
 
 
 
 
 
 
 
 
 
81
  {
82
  "id": "gpt-3.5-turbo",
83
  "description": "[openai/gpt-3.5-turbo]: https://platform.openai.com/docs/models/gpt-3-5-turbo",
 
6
  "gemma-7b": "google/gemma-1.1-7b-it",
7
  "command-r-plus": "CohereForAI/c4ai-command-r-plus",
8
  "llama3-70b": "meta-llama/Meta-Llama-3-70B-Instruct",
9
+ "zephyr-141b": "HuggingFaceH4/zephyr-orpo-141b-A35b-v0.1",
10
  "default": "mistralai/Mixtral-8x7B-Instruct-v0.1",
11
  }
12
 
 
27
  "mistral-7b": 32768,
28
  "openchat-3.5": 8192,
29
  "gemma-7b": 8192,
 
30
  "command-r-plus": 32768,
31
  "llama3-70b": 8192,
32
+ "zephyr-141b": 2048,
33
+ "gpt-3.5-turbo": 8192,
34
  }
35
 
36
  TOKEN_RESERVED = 20
 
73
  "created": 1700000000,
74
  "owned_by": "Google",
75
  },
76
+ {
77
+ "id": "command-r-plus",
78
+ "description": "[CohereForAI/c4ai-command-r-plus]: https://huggingface.co/CohereForAI/c4ai-command-r-plus",
79
+ "object": "model",
80
+ "created": 1700000000,
81
+ "owned_by": "CohereForAI",
82
+ },
83
+ {
84
+ "id": "llama3-70b",
85
+ "description": "[meta-llama/Meta-Llama-3-70B]: https://huggingface.co/meta-llama/Meta-Llama-3-70B",
86
+ "object": "model",
87
+ "created": 1700000000,
88
+ "owned_by": "Meta",
89
+ },
90
+ {
91
+ "id": "zephyr-141b",
92
+ "description": "[HuggingFaceH4/zephyr-orpo-141b-A35b-v0.1]: https://huggingface.co/HuggingFaceH4/zephyr-orpo-141b-A35b-v0.1",
93
+ "object": "model",
94
+ "created": 1700000000,
95
+ "owned_by": "Huggingface",
96
+ },
97
  {
98
  "id": "gpt-3.5-turbo",
99
  "description": "[openai/gpt-3.5-turbo]: https://platform.openai.com/docs/models/gpt-3-5-turbo",