From 933e970cb53948097938719569c99b917e706a03 Mon Sep 17 00:00:00 2001 From: Wing Lian Date: Fri, 26 May 2023 17:51:17 -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 6721537c2..cf5077134 100644 --- a/src/axolotl/utils/models.py +++ b/src/axolotl/utils/models.py @@ -41,7 +41,7 @@ def load_tokenizer( if tokenizer_type: tokenizer = getattr(transformers, tokenizer_type).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, ) else: tokenizer = AutoTokenizer.from_pretrained(