Update space

messagers/message_composer.py

@@ -102,12 +102,12 @@ class MessageComposer:
                 role = message["role"]
                 content = message["content"]
                 if role in self.inst_roles:
-                    self.cached_str = f"[INST] {content} [/INST]"
+                    self.cached_str = f"[/USER] {content} [/ASSIST]"
                 elif role in self.answer_roles:
                     self.merged_str += f"<s> {self.cached_str} {content} </s>\n"
                     self.cached_str = ""
                 else:
-                    self.cached_str = f"[INST] {content} [/INST]"
+                    self.cached_str = f"[/USER] {content} [/ASSIST]"
             if self.cached_str:
                 self.merged_str += f"{self.cached_str}"
         else:
@@ -187,14 +187,14 @@ class MessageComposer:
                 self.messages.append({"role": role, "content": content.strip()})
         elif self.model in ["zephyr-7b-beta"]:
             pair_pattern = (
-                r"<s>\s*\[INST\](?P<inst>[\s\S]*?)\[/INST\](?P<answer>[\s\S]*?)</s>"
+                r"<s>\s*\[INST\](?P<inst>[\s\S]*?)\[/USER\][/ASSIST\](?P<answer>[\s\S]*?)</s>"
             )
             pair_matches = re.finditer(pair_pattern, self.merged_str, re.MULTILINE)
             pair_matches_list = list(pair_matches)
 
             self.messages = self.convert_pair_matches_to_messages(pair_matches_list)
 
-            inst_pattern = r"\[INST\](?P<inst>[\s\S]*?)\[/INST\]"
+            inst_pattern = r"\[INST\](?P<inst>[\s\S]*?)\[/USER\][/ASSIST\]"
             inst_matches = re.finditer(inst_pattern, self.merged_str, re.MULTILINE)
             inst_matches_list = list(inst_matches)