From 145b060cbe9839220bdccdc4448a5e5df61a48a3 Mon Sep 17 00:00:00 2001 From: NanoCode012 Date: Mon, 29 May 2023 13:57:09 +0900 Subject: [PATCH] Lint alpaca_instruct --- src/axolotl/prompt_strategies/alpaca_instruct.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/src/axolotl/prompt_strategies/alpaca_instruct.py b/src/axolotl/prompt_strategies/alpaca_instruct.py index 6bce47ccd..0d0b267a6 100644 --- a/src/axolotl/prompt_strategies/alpaca_instruct.py +++ b/src/axolotl/prompt_strategies/alpaca_instruct.py @@ -1,10 +1,12 @@ +"""Module loading the AlpacaInstructPromptTokenizingStrategy class""" + from axolotl.prompt_tokenizers import AlpacaPromptTokenizingStrategy from axolotl.prompters import AlpacaPrompter, PromptStyle def load(tokenizer, cfg): return AlpacaPromptTokenizingStrategy( - AlpacaPrompter(PromptStyle.instruct), + AlpacaPrompter(PromptStyle.INSTRUCT), tokenizer, cfg.train_on_inputs, cfg.sequence_len,