ofirzaf commited on
Commit
a04960b
·
1 Parent(s): 07de8f2

Add OASST Falcon-7B conversation prompt

Browse files
Files changed (1) hide show
  1. conversation.py +14 -0
conversation.py CHANGED
@@ -432,6 +432,20 @@ register_conv_template(
432
  )
433
  )
434
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
435
  if __name__ == "__main__":
436
  conv = get_conv_template("vicuna_v1.1")
437
  conv.append_message(conv.roles[0], "Hello!")
 
432
  )
433
  )
434
 
435
+ # OpenAssistant Falcon-7B
436
+ register_conv_template(
437
+ Conversation(
438
+ name="falcon",
439
+ system="",
440
+ messages=(),
441
+ offset=0,
442
+ roles=("<|prompter|>", "<|assistant|>"),
443
+ sep_style=SeparatorStyle.NO_COLON_SINGLE,
444
+ sep="<|endoftext|>",
445
+ )
446
+ )
447
+
448
+
449
  if __name__ == "__main__":
450
  conv = get_conv_template("vicuna_v1.1")
451
  conv.append_message(conv.roles[0], "Hello!")