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,
|