fixes w/ example for super basic lora starter

2023-05-25 11:59:08 -04:00
parent 951facbb1f
commit a5d739b66b
3 changed files with 74 additions and 5 deletions
--- a/src/axolotl/prompters.py
+++ b/src/axolotl/prompters.py
@@ -18,7 +18,7 @@ class AlpacaPrompter:
    prompt_style = None

    def __init__(self, prompt_style="instruct"):
-        self.prompt_style = prompt_style
+        self.prompt_style = prompt_style if prompt_style else PromptStyle.instruct.value
        self.match_prompt_style()

    def match_prompt_style(self):