theluyuan/RWKV-Runner
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

change default World series prefix to User/Assistant

Browse Source
This commit is contained in:
josc146 2023-10-26 16:58:53 +08:00
parent 627a20936d
commit 43cfab5d4b
2 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
backend-python/utils/rwkv.py
View File

@ -356,8 +356,8 @@ class TextRWKV(AbstractRWKV):
self.END_OF_LINE = 187 self.END_OF_LINE = 187
else: else:
self.rwkv_type = RWKVType.World self.rwkv_type = RWKVType.World
self.user = "Question" self.user = "User"
self.bot = "Answer" self.bot = "Assistant"
self.END_OF_LINE = 11 self.END_OF_LINE = 11
self.AVOID_REPEAT_TOKENS = [] self.AVOID_REPEAT_TOKENS = []

4
frontend/src/pages/Chat.tsx
View File

@ -417,8 +417,8 @@ const ChatPanel: FC = observer(() => {
onClick={() => { onClick={() => {
let savedContent: string = ''; let savedContent: string = '';
const isWorldModel = commonStore.getCurrentModelConfig().modelParameters.modelName.toLowerCase().includes('world'); const isWorldModel = commonStore.getCurrentModelConfig().modelParameters.modelName.toLowerCase().includes('world');
const user = isWorldModel ? 'Question' : 'Bob'; const user = isWorldModel ? 'User' : 'Bob';
const bot = isWorldModel ? 'Answer' : 'Alice'; const bot = isWorldModel ? 'Assistant' : 'Alice';
commonStore.conversationOrder.forEach((uuid) => { commonStore.conversationOrder.forEach((uuid) => {
if (uuid === welcomeUuid) if (uuid === welcomeUuid)
return; return;