Nanobit commited on
Commit
eb480df
1 Parent(s): 133e676

Fix: ValueError when FA + Mistral when padding_side=right (#681)

Browse files

* Fix: ValueError when FA + Mistral when padding_side=right

* fix: remove tokenizer class check

Files changed (1) hide show
  1. src/axolotl/utils/models.py +3 -0
src/axolotl/utils/models.py CHANGED
@@ -81,6 +81,9 @@ def load_tokenizer(cfg):
81
  tokenizer.add_special_tokens({"pad_token": "[PAD]"})
82
  os.environ["TOKENIZERS_PARALLELISM"] = "false"
83
 
 
 
 
84
  if cfg.special_tokens:
85
  for k, val in cfg.special_tokens.items():
86
  tokenizer.add_special_tokens(
 
81
  tokenizer.add_special_tokens({"pad_token": "[PAD]"})
82
  os.environ["TOKENIZERS_PARALLELISM"] = "false"
83
 
84
+ if cfg.is_mistral_derived_model:
85
+ tokenizer.padding_side = "left"
86
+
87
  if cfg.special_tokens:
88
  for k, val in cfg.special_tokens.items():
89
  tokenizer.add_special_tokens(