ka1kuk commited on
Commit
3895485
β€’
1 Parent(s): e1cd27e

Update networks/message_streamer.py

Browse files
Files changed (1) hide show
  1. networks/message_streamer.py +1 -5
networks/message_streamer.py CHANGED
@@ -13,11 +13,7 @@ class MessageStreamer:
13
  "mistral-7b": "mistralai/Mistral-7B-Instruct-v0.2", # 65.71, fast
14
  "nous-mixtral-8x7b": "NousResearch/Nous-Hermes-2-Mixtral-8x7B-DPO",
15
  "gemma-7b": "google/gemma-7b-it",
16
- # "openchat-3.5": "openchat/openchat-3.5-1210", # 68.89, fast
17
- # "zephyr-7b-beta": "HuggingFaceH4/zephyr-7b-beta", # ❌ Too Slow
18
- # "llama-70b": "meta-llama/Llama-2-70b-chat-hf", # ❌ Require Pro User
19
- # "codellama-34b": "codellama/CodeLlama-34b-Instruct-hf", # ❌ Low Score
20
- # "falcon-180b": "tiiuae/falcon-180B-chat", # ❌ Require Pro User
21
  "default": "mistralai/Mixtral-8x7B-Instruct-v0.1",
22
  }
23
  STOP_SEQUENCES_MAP = {
 
13
  "mistral-7b": "mistralai/Mistral-7B-Instruct-v0.2", # 65.71, fast
14
  "nous-mixtral-8x7b": "NousResearch/Nous-Hermes-2-Mixtral-8x7B-DPO",
15
  "gemma-7b": "google/gemma-7b-it",
16
+ "codellama-7b": "codellama/CodeLlama-7b-hf", # ❌ Low Score
 
 
 
 
17
  "default": "mistralai/Mixtral-8x7B-Instruct-v0.1",
18
  }
19
  STOP_SEQUENCES_MAP = {