mmarch/text-generation-webui
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Fix loading characters

Browse source
This commit is contained in:
oobabooga 2023-04-05 14:22:32 -03:00
parent 90141bc1a8
commit 7f66421369
3 changed files with 8 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

8
modules/chat.py
View file

@ -332,7 +332,7 @@ def generate_pfp_cache(character):
return img
return None
def load_character(character, name1, name2, instruct=False):
def load_character(character, name1, name2, mode):
shared.character = character
shared.history['internal'] = []
shared.history['visible'] = []
@ -345,7 +345,7 @@ def load_character(character, name1, name2, instruct=False):
Path("cache/pfp_character.png").unlink()
if character != 'None':
folder = "characters" if not instruct else "characters/instruction-following"
folder = 'characters' if not mode == 'instruct' else 'characters/instruction-following'
picture = generate_pfp_cache(character)
for extension in ["yml", "yaml", "json"]:
filepath = Path(f'{folder}/{character}.{extension}')
@ -386,10 +386,10 @@ def load_character(character, name1, name2, instruct=False):
shared.history['internal'] += [['<|BEGIN-VISIBLE-CHAT|>', greeting]]
shared.history['visible'] += [['', apply_extensions(greeting, "output")]]
return name1, name2, picture, greeting, context, end_of_turn, chat_html_wrapper(shared.history['visible'], name1, name2, reset_cache=True)
return name1, name2, picture, greeting, context, end_of_turn, chat_html_wrapper(shared.history['visible'], name1, name2, mode, reset_cache=True)
def load_default_history(name1, name2):
load_character("None", name1, name2)
load_character("None", name1, name2, "chat")
def upload_character(json_file, img, tavern=False):
json_file = json_file if type(json_file) == str else json_file.decode('utf-8')

3
modules/html_generator.py
View file

@ -203,8 +203,7 @@ def generate_cai_chat_html(history, name1, name2, reset_cache=False):
def generate_chat_html(history, name1, name2):
return generate_cai_chat_html(history, name1, name2)
def chat_html_wrapper(history, name1, name2, mode="cai-chat", reset_cache=False):
def chat_html_wrapper(history, name1, name2, mode, reset_cache=False):
if mode == "cai-chat":
return generate_cai_chat_html(history, name1, name2, reset_cache)
elif mode == "chat":