winglian commited on
Commit
2c66a20
1 Parent(s): d3e686c

add lmsys vicuna to the arena

Browse files
Files changed (1) hide show
  1. app.py +1 -2
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
  }