Minor fixes

This commit is contained in:
oobabooga 2023-01-19 19:04:54 -03:00
parent 2181fca709
commit e61138bdad
2 changed files with 8 additions and 6 deletions

View File

@ -333,9 +333,9 @@ if args.chat or args.cai_chat:
context += f"{data['char_name']}'s Persona: {data['char_persona']}\n" context += f"{data['char_name']}'s Persona: {data['char_persona']}\n"
if 'world_scenario' in data and data['world_scenario'] != '': if 'world_scenario' in data and data['world_scenario'] != '':
context += f"Scenario: {data['world_scenario']}\n" context += f"Scenario: {data['world_scenario']}\n"
context = f"{context.strip()}\n<START>" context = f"{context.strip()}\n<START>\n"
if 'example_dialogue' in data and data['example_dialogue'] != '': if 'example_dialogue' in data and data['example_dialogue'] != '':
context += f"{data['example_dialogue']}" context += f"{data['example_dialogue'].strip()}\n"
if 'char_greeting' in data: if 'char_greeting' in data:
history = [['', data['char_greeting']]] history = [['', data['char_greeting']]]
else: else:

View File

@ -5,11 +5,13 @@
"preset": "NovelAI-Sphinx Moth", "preset": "NovelAI-Sphinx Moth",
"name1": "Person 1", "name1": "Person 1",
"name2": "Person 2", "name2": "Person 2",
"name1_pygmalion": "You",
"name2_pygmalion": "Kawaii",
"context": "This is a conversation between two people.", "context": "This is a conversation between two people.",
"context_pygmalion": "This is a conversation between two people.\n<START>",
"prompt": "Common sense questions and answers\n\nQuestion: \nFactual answer:", "prompt": "Common sense questions and answers\n\nQuestion: \nFactual answer:",
"prompt_gpt4chan": "-----\n--- 865467536\nInput text\n--- 865467537\n", "prompt_gpt4chan": "-----\n--- 865467536\nInput text\n--- 865467537\n",
"stop_at_newline": true "stop_at_newline": true,
"preset_pygmalion": "Pygmalion",
"name1_pygmalion": "You",
"name2_pygmalion": "Kawaii",
"context_pygmalion": "This is a conversation between two people.\n<START>",
"stop_at_newline_pygmalion": false
} }