Hansimov commited on
Commit
85354a9
1 Parent(s): cec776f

:pencil: [Doc] Available models: More comments on dropped models

Browse files
Files changed (1) hide show
  1. networks/message_streamer.py +5 -3
networks/message_streamer.py CHANGED
@@ -10,9 +10,11 @@ class MessageStreamer:
10
  MODEL_MAP = {
11
  "mixtral-8x7b": "mistralai/Mixtral-8x7B-Instruct-v0.1", # 72.62, fast [Recommended]
12
  "mistral-7b": "mistralai/Mistral-7B-Instruct-v0.2", # 65.71, fast
13
- "openchat-3.5": "openchat/openchat-3.5-1210", # ??, fast
14
- # "zephyr-7b-alpha": "HuggingFaceH4/zephyr-7b-alpha", # 59.5, fast
15
- # "zephyr-7b-beta": "HuggingFaceH4/zephyr-7b-beta", # 61.95, slow
 
 
16
  "default": "mistralai/Mixtral-8x7B-Instruct-v0.1",
17
  }
18
  STOP_SEQUENCES_MAP = {
 
10
  MODEL_MAP = {
11
  "mixtral-8x7b": "mistralai/Mixtral-8x7B-Instruct-v0.1", # 72.62, fast [Recommended]
12
  "mistral-7b": "mistralai/Mistral-7B-Instruct-v0.2", # 65.71, fast
13
+ "openchat-3.5": "openchat/openchat-3.5-1210", # 68.89, fast
14
+ # "zephyr-7b-beta": "HuggingFaceH4/zephyr-7b-beta", # Too Slow
15
+ # "llama-70b": "meta-llama/Llama-2-70b-chat-hf", # Require Pro User
16
+ # "codellama-34b": "codellama/CodeLlama-34b-Instruct-hf", # ❌ Low Score
17
+ # "falcon-180b": "tiiuae/falcon-180B-chat", # ❌ Require Pro User
18
  "default": "mistralai/Mixtral-8x7B-Instruct-v0.1",
19
  }
20
  STOP_SEQUENCES_MAP = {