Spaces:
Sleeping
Sleeping
Commit
·
c8005a5
1
Parent(s):
9f857bc
Update app.py
Browse files
app.py
CHANGED
@@ -49,8 +49,8 @@ def mmlu_display_question_answer(question, cot, request: gr.Request):
|
|
49 |
|
50 |
return q["agent_response"]["llama"][0], q["agent_response"]["wizardlm"][0], q["agent_response"]["orca"][0], q["summarization"][0], q["agent_response"]["llama"][1], q["agent_response"]["wizardlm"][1], q["agent_response"]["orca"][1], q["summarization"][1], q["agent_response"]["llama"][2], q["agent_response"]["wizardlm"][2], q["agent_response"]["orca"][2]
|
51 |
|
52 |
-
|
53 |
-
def warmup(model_list=
|
54 |
for i in range(len(model_list)):
|
55 |
API_URL = model_inference_endpoints[model_list[i]["API_URL"]]
|
56 |
headers = model_inference_endpoints[model_list[i]["headers"]]
|
|
|
49 |
|
50 |
return q["agent_response"]["llama"][0], q["agent_response"]["wizardlm"][0], q["agent_response"]["orca"][0], q["summarization"][0], q["agent_response"]["llama"][1], q["agent_response"]["wizardlm"][1], q["agent_response"]["orca"][1], q["summarization"][1], q["agent_response"]["llama"][2], q["agent_response"]["wizardlm"][2], q["agent_response"]["orca"][2]
|
51 |
|
52 |
+
warmup_test = ["llama"]
|
53 |
+
def warmup(model_list=warmup_test, model_inference_endpoints=inference_endpoint):
|
54 |
for i in range(len(model_list)):
|
55 |
API_URL = model_inference_endpoints[model_list[i]["API_URL"]]
|
56 |
headers = model_inference_endpoints[model_list[i]["headers"]]
|