Mirrors/text-generation-webui
1
0
Fork 0
mirror of https://github.com/oobabooga/text-generation-webui.git synced 2024-11-26 01:30:20 +01:00
Code Issues Packages Projects Releases Wiki Activity

Force your name to be "You" for pygmalion

Browse Source 
This allows you to customize your displayed name.
This commit is contained in:
oobabooga 2023-02-15 21:32:53 -03:00
parent 3746d72853
commit 7bd2ae05bf
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
server.py
View File

@ -505,8 +505,10 @@ def clean_chat_message(text):
return text return text
def generate_chat_prompt(text, tokens, name1, name2, context, chat_prompt_size, impersonate=False): def generate_chat_prompt(text, tokens, name1, name2, context, chat_prompt_size, impersonate=False):
text = clean_chat_message(text) if 'pygmalion' in model_name.lower():
name1 = "You"
text = clean_chat_message(text)
rows = [f"{context.strip()}\n"] rows = [f"{context.strip()}\n"]
i = len(history['internal'])-1 i = len(history['internal'])-1
count = 0 count = 0