winglian commited on
Commit
fb100a9
·
1 Parent(s): 3a50377

fix enum pass as value

Browse files
src/axolotl/prompt_strategies/alpaca_chat.py CHANGED
@@ -4,7 +4,7 @@ from axolotl.prompters import AlpacaPrompter, PromptStyle
4
 
5
  def load(tokenizer, cfg):
6
  return AlpacaPromptTokenizingStrategy(
7
- AlpacaPrompter(PromptStyle.chat), tokenizer, cfg.train_on_inputs, cfg.sequence_len
8
  )
9
 
10
 
@@ -19,5 +19,5 @@ class AlpacaQAPromptTokenizingStrategy(InstructionPromptTokenizingStrategy):
19
 
20
  def load_qa(tokenizer, cfg):
21
  return AlpacaQAPromptTokenizingStrategy(
22
- AlpacaPrompter(PromptStyle.chat), tokenizer, cfg.train_on_inputs, cfg.sequence_len
23
  )
 
4
 
5
  def load(tokenizer, cfg):
6
  return AlpacaPromptTokenizingStrategy(
7
+ AlpacaPrompter(PromptStyle.chat.value), tokenizer, cfg.train_on_inputs, cfg.sequence_len
8
  )
9
 
10
 
 
19
 
20
  def load_qa(tokenizer, cfg):
21
  return AlpacaQAPromptTokenizingStrategy(
22
+ AlpacaPrompter(PromptStyle.chat.value), tokenizer, cfg.train_on_inputs, cfg.sequence_len
23
  )