change default World series prefix to User/Assistant

2023-10-26 16:58:53 +08:00
parent 627a20936d
commit 43cfab5d4b
2 changed files with 4 additions and 4 deletions
--- a/backend-python/utils/rwkv.py
+++ b/backend-python/utils/rwkv.py
@@ -356,8 +356,8 @@ class TextRWKV(AbstractRWKV):
            self.END_OF_LINE = 187
        else:
            self.rwkv_type = RWKVType.World
-            self.user = "Question"
-            self.bot = "Answer"
+            self.user = "User"
+            self.bot = "Assistant"
            self.END_OF_LINE = 11

        self.AVOID_REPEAT_TOKENS = []