Spaces:
Runtime error
Runtime error
add lmsys vicuna to the arena
Browse files
app.py
CHANGED
@@ -93,8 +93,6 @@ class Pipeline:
|
|
93 |
return
|
94 |
elif response.status_code >= 400:
|
95 |
logging.error(response.json())
|
96 |
-
# Sleep for 0.5 seconds between each request
|
97 |
-
sleep(0.5)
|
98 |
|
99 |
def poll_for_status(self, task_id):
|
100 |
url = f"https://api.runpod.ai/v2/{self.endpoint_id}/status/{task_id}"
|
@@ -122,6 +120,7 @@ AVAILABLE_MODELS = {
|
|
122 |
"manticore-13b-chat": ("u6tv84bpomhfei", prompt_chat),
|
123 |
"airoboros-13b": ("rglzxnk80660ja", prompt_chat),
|
124 |
"wizard-vicuna-13b": ("9vvpikt4ttyqos", prompt_chat),
|
|
|
125 |
"supercot-13b": ("0be7865dwxpwqk", prompt_instruct, ["Instruction:"]),
|
126 |
"mpt-7b-instruct": ("jpqbvnyluj18b0", prompt_instruct),
|
127 |
}
|
|
|
93 |
return
|
94 |
elif response.status_code >= 400:
|
95 |
logging.error(response.json())
|
|
|
|
|
96 |
|
97 |
def poll_for_status(self, task_id):
|
98 |
url = f"https://api.runpod.ai/v2/{self.endpoint_id}/status/{task_id}"
|
|
|
120 |
"manticore-13b-chat": ("u6tv84bpomhfei", prompt_chat),
|
121 |
"airoboros-13b": ("rglzxnk80660ja", prompt_chat),
|
122 |
"wizard-vicuna-13b": ("9vvpikt4ttyqos", prompt_chat),
|
123 |
+
"lmsys-vicuna-13b": ("2nlb32ydkaz6yd", prompt_chat),
|
124 |
"supercot-13b": ("0be7865dwxpwqk", prompt_instruct, ["Instruction:"]),
|
125 |
"mpt-7b-instruct": ("jpqbvnyluj18b0", prompt_instruct),
|
126 |
}
|