From 004820209d5f1954f11091bbdc7c84fab77614b2 Mon Sep 17 00:00:00 2001 From: Wing Lian Date: Thu, 25 May 2023 12:21:02 -0400 Subject: [PATCH] Update src/axolotl/prompters.py Co-authored-by: NanoCode012 --- src/axolotl/prompters.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/axolotl/prompters.py b/src/axolotl/prompters.py index df37ec85a..fd9dfc8d4 100644 --- a/src/axolotl/prompters.py +++ b/src/axolotl/prompters.py @@ -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()