Spaces:
Running
Running
fix the typo of BF16
Browse files- app.py +1 -1
- constant.py +2 -2
app.py
CHANGED
@@ -184,7 +184,7 @@ with gr.Blocks(gr.themes.Soft(), js=js_code_label, css=my_css) as demo:
|
|
184 |
rp_2 = gr.Slider(label="Repetition Penalty", step=0.1, minimum=0.1, maximum=2.0, value=1.0)
|
185 |
|
186 |
left_model_choice.value = "Llama-3.1-405B-FP8"
|
187 |
-
right_model_choice.value = "Llama-3.1-405B-Instruct-
|
188 |
left_model_choice.change(load_models, [left_model_choice], [chat_a, chat_b, right_model_choice])
|
189 |
|
190 |
model_type_left = gr.Textbox(visible=False, value="base")
|
|
|
184 |
rp_2 = gr.Slider(label="Repetition Penalty", step=0.1, minimum=0.1, maximum=2.0, value=1.0)
|
185 |
|
186 |
left_model_choice.value = "Llama-3.1-405B-FP8"
|
187 |
+
right_model_choice.value = "Llama-3.1-405B-Instruct-BF16"
|
188 |
left_model_choice.change(load_models, [left_model_choice], [chat_a, chat_b, right_model_choice])
|
189 |
|
190 |
model_type_left = gr.Textbox(visible=False, value="base")
|
constant.py
CHANGED
@@ -49,7 +49,7 @@ BASE_TO_ALIGNED = {
|
|
49 |
"Mistral-7B-v0.1": "Mistral-7B-v0.1-Instruct",
|
50 |
"Mixtral-8x22B": "Mixtral-8x22B-Instruct",
|
51 |
"Qwen1.5-72B": "Qwen1.5-72B-Instruct",
|
52 |
-
"Llama-3.1-405B-FP8": "Llama-3.1-405B-Instruct-
|
53 |
"Yi-34B": "Yi-34B-chat",
|
54 |
"Llama-2-7B": "Llama-2-7B-chat",
|
55 |
"Llama-2-70B": "Llama-2-70B-chat",
|
@@ -79,7 +79,7 @@ MODEL_MAPPING = {
|
|
79 |
"Llama-2-7B-chat": "meta-llama/Llama-2-7b-chat-hf",
|
80 |
"Llama-2-70B-chat": "meta-llama/Llama-2-70b-chat-hf",
|
81 |
"OLMo-7B-Instruct": "allenai/OLMo-7B-Instruct",
|
82 |
-
"Llama-3.1-405B-Instruct-
|
83 |
}
|
84 |
|
85 |
|
|
|
49 |
"Mistral-7B-v0.1": "Mistral-7B-v0.1-Instruct",
|
50 |
"Mixtral-8x22B": "Mixtral-8x22B-Instruct",
|
51 |
"Qwen1.5-72B": "Qwen1.5-72B-Instruct",
|
52 |
+
"Llama-3.1-405B-FP8": "Llama-3.1-405B-Instruct-BF16",
|
53 |
"Yi-34B": "Yi-34B-chat",
|
54 |
"Llama-2-7B": "Llama-2-7B-chat",
|
55 |
"Llama-2-70B": "Llama-2-70B-chat",
|
|
|
79 |
"Llama-2-7B-chat": "meta-llama/Llama-2-7b-chat-hf",
|
80 |
"Llama-2-70B-chat": "meta-llama/Llama-2-70b-chat-hf",
|
81 |
"OLMo-7B-Instruct": "allenai/OLMo-7B-Instruct",
|
82 |
+
"Llama-3.1-405B-Instruct-BF16": "meta-llama/Meta-Llama-3.1-405B-Instruct",
|
83 |
}
|
84 |
|
85 |
|