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

fix robin's prompt

Browse files
Files changed (1) hide show
  1. app.py +7 -1
app.py CHANGED
@@ -28,6 +28,12 @@ dynamodb = boto3.resource('dynamodb', region_name='us-east-1')
28
  table = dynamodb.Table('oaaic_chatbot_arena')
29
 
30
 
 
 
 
 
 
 
31
  def prompt_instruct(system_msg, history):
32
  return system_msg.strip() + "\n" + \
33
  "\n".join(["\n".join(["### Instruction: "+item[0], "### Response: "+item[1]])
@@ -141,7 +147,7 @@ AVAILABLE_MODELS = {
141
  "minotaur-13b-fixed": ("sjnkstd3e40ojj", prompt_roleplay),
142
  "wizardlm-13b": ("k0chcxsgukov8x", prompt_instruct),
143
  "selfee-13b": ("50rnvxln9bmf4c", prompt_instruct),
144
- "robin-v2-13b": ("4cw4vwzzhsl5pq", prompt_instruct),
145
  }
146
 
147
  OAAIC_MODELS = [
 
28
  table = dynamodb.Table('oaaic_chatbot_arena')
29
 
30
 
31
+ def prompt_human_instruct(system_msg, history):
32
+ return system_msg.strip() + "\n" + \
33
+ "\n".join(["\n".join(["###Human: "+item[0], "###Assistant: "+item[1]])
34
+ for item in history])
35
+
36
+
37
  def prompt_instruct(system_msg, history):
38
  return system_msg.strip() + "\n" + \
39
  "\n".join(["\n".join(["### Instruction: "+item[0], "### Response: "+item[1]])
 
147
  "minotaur-13b-fixed": ("sjnkstd3e40ojj", prompt_roleplay),
148
  "wizardlm-13b": ("k0chcxsgukov8x", prompt_instruct),
149
  "selfee-13b": ("50rnvxln9bmf4c", prompt_instruct),
150
+ "robin-v2-13b": ("4cw4vwzzhsl5pq", prompt_human_instruct, ["###Human"]),
151
  }
152
 
153
  OAAIC_MODELS = [