qwerrwe / src /axolotl /prompt_strategies /alpaca_instruct.py
winglian's picture
apply black formatting
ce34d64
raw
history blame
324 Bytes
from axolotl.prompt_tokenizers import AlpacaPromptTokenizingStrategy
from axolotl.prompters import AlpacaPrompter, PromptStyle
def load(tokenizer, cfg):
return AlpacaPromptTokenizingStrategy(
AlpacaPrompter(PromptStyle.instruct),
tokenizer,
cfg.train_on_inputs,
cfg.sequence_len,
)