diff --git a/src/axolotl/utils/models.py b/src/axolotl/utils/models.py index 45e771deb..80707ff8a 100644 --- a/src/axolotl/utils/models.py +++ b/src/axolotl/utils/models.py @@ -167,7 +167,7 @@ def load_model( else: tokenizer = getattr(transformers, tokenizer_type).from_pretrained(model) except: - tokenizer = AutoTokenizer.from_pretrained(base_model_config) + tokenizer = AutoTokenizer.from_pretrained(base_model_config, trust_remote_code=True if cfg.trust_remote_code is True else False) logging.debug(f"EOS: {tokenizer.eos_token_id} / {tokenizer.eos_token}") logging.debug(f"BOS: {tokenizer.bos_token_id} / {tokenizer.bos_token}")