AI/text-generation-webui
1
0
Fork 0
mirror of https://github.com/oobabooga/text-generation-webui.git synced 2024-09-19 15:56:56 +00: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

5
modules/chat.py
View File

@ -349,11 +349,12 @@ def load_character(_character, name1, name2):
if _character != 'None': if _character != 'None':
shared.character = _character shared.character = _character
for extension in ["yml", "yaml", "json"]: for extension in ["yml", "yaml", "json"]:
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']: