mmarch/text-generation-webui
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages 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
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