Lint alpaca_instruct
Browse files
src/axolotl/prompt_strategies/alpaca_instruct.py
CHANGED
@@ -1,10 +1,12 @@
|
|
|
|
|
|
1 |
from axolotl.prompt_tokenizers import AlpacaPromptTokenizingStrategy
|
2 |
from axolotl.prompters import AlpacaPrompter, PromptStyle
|
3 |
|
4 |
|
5 |
def load(tokenizer, cfg):
|
6 |
return AlpacaPromptTokenizingStrategy(
|
7 |
-
AlpacaPrompter(PromptStyle.
|
8 |
tokenizer,
|
9 |
cfg.train_on_inputs,
|
10 |
cfg.sequence_len,
|
|
|
1 |
+
"""Module loading the AlpacaInstructPromptTokenizingStrategy class"""
|
2 |
+
|
3 |
from axolotl.prompt_tokenizers import AlpacaPromptTokenizingStrategy
|
4 |
from axolotl.prompters import AlpacaPrompter, PromptStyle
|
5 |
|
6 |
|
7 |
def load(tokenizer, cfg):
|
8 |
return AlpacaPromptTokenizingStrategy(
|
9 |
+
AlpacaPrompter(PromptStyle.INSTRUCT),
|
10 |
tokenizer,
|
11 |
cfg.train_on_inputs,
|
12 |
cfg.sequence_len,
|