From ca1bb92337951622832e0be0d2158c315a940054 Mon Sep 17 00:00:00 2001 From: Wing Lian Date: Fri, 26 May 2023 17:51:24 -0400 Subject: [PATCH] Update src/axolotl/utils/models.py Co-authored-by: NanoCode012 --- src/axolotl/utils/models.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/axolotl/utils/models.py b/src/axolotl/utils/models.py index cf5077134..d73776e76 100644 --- a/src/axolotl/utils/models.py +++ b/src/axolotl/utils/models.py @@ -46,7 +46,7 @@ def load_tokenizer( else: tokenizer = AutoTokenizer.from_pretrained( base_model_config, - trust_remote_code=True if cfg.trust_remote_code is True else False, + trust_remote_code=cfg.trust_remote_code or False, ) logging.debug(f"EOS: {tokenizer.eos_token_id} / {tokenizer.eos_token}")