Add OASST Falcon-7B conversation prompt
Browse files- 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!")
|