winglian commited on
Commit
1fb15cc
1 Parent(s): dc03939

bootstrap robin to get more battles

Browse files
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -207,6 +207,9 @@ def chat(history1, history2, system_msg, state):
207
  if not arena_bots:
208
  arena_bots = list(AVAILABLE_MODELS.keys())
209
  random.shuffle(arena_bots)
 
 
 
210
 
211
  battle = arena_bots[0:2]
212
  model1 = get_model_pipeline(battle[0])
 
207
  if not arena_bots:
208
  arena_bots = list(AVAILABLE_MODELS.keys())
209
  random.shuffle(arena_bots)
210
+ # bootstrap a new bot into the arena more often
211
+ if "robin-v2-13b" not in arena_bots[0:2] and random.choice([True, False]):
212
+ arena_bots.insert(random.choice([0,1]), "robin-v2-13b")
213
 
214
  battle = arena_bots[0:2]
215
  model1 = get_model_pipeline(battle[0])