Fix a bug in the API (closes #4027)

This commit is contained in:
oobabooga 2023-09-21 17:54:42 -07:00
parent 05c4a4f83c
commit d6814d7c15

View File

@ -67,7 +67,7 @@ def build_parameters(body, chat=False):
if str(instruction_template) == "None": if str(instruction_template) == "None":
instruction_template = "Vicuna-v1.1" instruction_template = "Vicuna-v1.1"
name1, name2, _, greeting, context, _ = load_character_memoized(character, str(body.get('your_name', shared.settings['name1'])), shared.settings['name2'], instruct=False) name1, name2, _, greeting, context, _ = load_character_memoized(character, str(body.get('your_name', shared.settings['name1'])), '', instruct=False)
name1_instruct, name2_instruct, _, _, context_instruct, turn_template = load_character_memoized(instruction_template, '', '', instruct=True) name1_instruct, name2_instruct, _, _, context_instruct, turn_template = load_character_memoized(instruction_template, '', '', instruct=True)
generate_params.update({ generate_params.update({
'mode': str(body.get('mode', 'chat')), 'mode': str(body.get('mode', 'chat')),