Update src/axolotl/prompters.py

Co-authored-by: NanoCode012 <kevinvong@rocketmail.com>
This commit is contained in:
Wing Lian
2023-05-25 12:21:02 -04:00
committed by GitHub
parent e396654319
commit 004820209d

View File

@@ -17,7 +17,7 @@ class AlpacaPrompter:
system_no_input_prompt = "Below is an instruction that describes a task. Write a response that appropriately completes the request.\n\n"
prompt_style = None
def __init__(self, prompt_style="instruct"):
def __init__(self, prompt_style=PromptStyle.instruct.value):
self.prompt_style = prompt_style if prompt_style else PromptStyle.instruct.value
self.match_prompt_style()