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 = {
|