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"
if 'world_scenario' in data and data['world_scenario'] != '':
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'] != '':
context += f"{data['example_dialogue']}"
context += f"{data['example_dialogue'].strip()}\n"
if 'char_greeting' in data:
history = [['', data['char_greeting']]]
else: