| import copy | |
| class Conversation: | |
| roles = ('<|im_start|>user', '<|im_start|>assistant') | |
| system = '<|im_start|>system\nYou are a helpful language and vision assistant. You are able to understand the visual content that the user provides, and assist the user with a variety of tasks using natural language.' | |
| def __init__(self): | |
| self.messages = [] | |
| def copy(self): | |
| return copy.deepcopy(self) | |
| def append_message(self, role, message): | |
| self.messages.append((role, message)) | |
| def get_prompt(self): | |
| text = self.system + '<|im_end|>\n' if self.system else '' | |
| for role, message in self.messages: | |
| text += role + '\n' | |
| if message: | |
| text += message + '<|im_end|>\n' | |
| return text | |
| conv_templates = {'ministral': Conversation()} | |