mirror of
https://github.com/oobabooga/text-generation-webui.git
synced 2024-12-24 13:28:59 +01:00
Fix some old JSON characters not loading (#740)
This commit is contained in:
parent
8b442305ac
commit
8b1f20aa04
@ -349,11 +349,12 @@ def load_character(_character, name1, name2):
|
||||
if _character != 'None':
|
||||
shared.character = _character
|
||||
|
||||
for extension in ["yml", "yaml", "json"]:
|
||||
for extension in ["yml", "yaml", "json"]:
|
||||
filepath = Path(f'characters/{_character}.{extension}')
|
||||
if filepath.exists():
|
||||
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']
|
||||
for field in ['context', 'greeting', 'example_dialogue', 'char_persona', 'char_greeting', 'world_scenario']:
|
||||
|
Loading…
Reference in New Issue
Block a user