Update networks/message_streamer.py
Browse files
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 |
-
|
| 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 = {
|