| 113 | |
| 114 | |
| 115 | class VicunaPrompter(BasePrompter): |
| 116 | def __init__(self): |
| 117 | system_inst = "A chat between a curious user and an artificial intelligence assistant. The assistant gives helpful, detailed, and polite answers to the user's questions." |
| 118 | role1 = "USER" |
| 119 | role2 = "ASSISTANT" |
| 120 | sen_spliter = " " |
| 121 | qa_spliter = "</s>" |
| 122 | super().__init__(system_inst, role1, role2, sen_spliter, qa_spliter) |
| 123 | |
| 124 | |
| 125 | class Llama2Prompter(OneShotBasePrompter): |