Nanobit commited on
Commit
145b060
1 Parent(s): 8cc0aad

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.instruct),
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,