Update space

messagers/message_composer.py

@@ -98,7 +98,7 @@ class MessageComposer:
         elif self.model in ["gpt-neo-2.7B"]:
             self.messages = self.concat_messages_by_role(messages)
             self.merged_str_list = []
-            self.end_of_turn = "<|endoftext|>"
+            self.end_of_turn = "<|end_of_turn|>"
             for message in self.messages:
                 role = message["role"]
                 content = message["content"]

networks/message_streamer.py

@@ -23,7 +23,7 @@ class MessageStreamer:
         "mixtral-8x7b": "</s>",
         "mistral-7b": "</s>",
         "nous-mixtral-8x7b": "<|im_end|>",
-        "gpt-neo-2.7B": "<|endoftext|>",
+        "gpt-neo-2.7B": "<|end_of_turn|>",
 
     }
     TOKEN_LIMIT_MAP = {