AI/text-generation-webui
1
0
Fork 0
mirror of https://github.com/oobabooga/text-generation-webui.git synced 2024-10-01 01:26:03 -04:00
Code Issues Packages Projects Releases Wiki Activity

Fix some old JSON characters not loading (#740)

Browse Source
This commit is contained in:
Alex "mcmonkey" Goodwin 2023-04-03 06:49:28 -07:00 committed by GitHub
parent 8b442305ac
commit 8b1f20aa04
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
modules/chat.py
View File

@ -353,7 +353,8 @@ def load_character(_character, name1, name2):
filepath = Path(f'characters/{_character}.{extension}') filepath = Path(f'characters/{_character}.{extension}')
if filepath.exists(): if filepath.exists():
break break
data = yaml.safe_load(open(filepath, 'r', encoding='utf-8').read()) file_contents = open(filepath, 'r', encoding='utf-8').read()
data = json.loads(file_contents) if extension == "json" else yaml.safe_load(file_contents)
name2 = data['name'] if 'name' in data else data['char_name'] name2 = data['name'] if 'name' in data else data['char_name']
for field in ['context', 'greeting', 'example_dialogue', 'char_persona', 'char_greeting', 'world_scenario']: for field in ['context', 'greeting', 'example_dialogue', 'char_persona', 'char_greeting', 'world_scenario']: