Hansimov commited on
Commit
06c8d58
1 Parent(s): fa3bfb0

:zap: [Enhance] Remove openchat-3.5 from available models

Browse files
Files changed (2) hide show
  1. apis/chat_api.py +4 -4
  2. networks/message_streamer.py +1 -1
apis/chat_api.py CHANGED
@@ -35,10 +35,10 @@ class ChatAPIApp:
35
  "id": "mistral-7b",
36
  "description": "[mistralai/Mistral-7B-Instruct-v0.2]: https://huggingface.co/mistralai/Mistral-7B-Instruct-v0.2",
37
  },
38
- {
39
- "id": "openchat-3.5",
40
- "description": "[openchat/openchat-3.5-1210]: https://huggingface.co/openchat/openchat-3.5-1210",
41
- },
42
  ]
43
  return self.available_models
44
 
 
35
  "id": "mistral-7b",
36
  "description": "[mistralai/Mistral-7B-Instruct-v0.2]: https://huggingface.co/mistralai/Mistral-7B-Instruct-v0.2",
37
  },
38
+ # {
39
+ # "id": "openchat-3.5",
40
+ # "description": "[openchat/openchat-3.5-1210]: https://huggingface.co/openchat/openchat-3.5-1210",
41
+ # },
42
  ]
43
  return self.available_models
44
 
networks/message_streamer.py CHANGED
@@ -11,7 +11,7 @@ class MessageStreamer:
11
  MODEL_MAP = {
12
  "mixtral-8x7b": "mistralai/Mixtral-8x7B-Instruct-v0.1", # 72.62, fast [Recommended]
13
  "mistral-7b": "mistralai/Mistral-7B-Instruct-v0.2", # 65.71, fast
14
- "openchat-3.5": "openchat/openchat-3.5-1210", # 68.89, fast
15
  # "zephyr-7b-beta": "HuggingFaceH4/zephyr-7b-beta", # ❌ Too Slow
16
  # "llama-70b": "meta-llama/Llama-2-70b-chat-hf", # ❌ Require Pro User
17
  # "codellama-34b": "codellama/CodeLlama-34b-Instruct-hf", # ❌ Low Score
 
11
  MODEL_MAP = {
12
  "mixtral-8x7b": "mistralai/Mixtral-8x7B-Instruct-v0.1", # 72.62, fast [Recommended]
13
  "mistral-7b": "mistralai/Mistral-7B-Instruct-v0.2", # 65.71, fast
14
+ # "openchat-3.5": "openchat/openchat-3.5-1210", # 68.89, fast
15
  # "zephyr-7b-beta": "HuggingFaceH4/zephyr-7b-beta", # ❌ Too Slow
16
  # "llama-70b": "meta-llama/Llama-2-70b-chat-hf", # ❌ Require Pro User
17
  # "codellama-34b": "codellama/CodeLlama-34b-Instruct-hf", # ❌ Low Score