tocmo0nlord/axolotl
1
0
Fork 0
Code Issues Pull Requests Actions 3 Packages Projects Releases Wiki Activity

Update src/axolotl/utils/models.py

Browse Source 
Co-authored-by: NanoCode012 <kevinvong@rocketmail.com>
This commit is contained in:
Wing Lian
2023-05-26 17:51:17 -04:00
committed by GitHub
parent 32e6fe9286
commit 933e970cb5
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/axolotl/utils/models.py
View File

@@ -41,7 +41,7 @@ def load_tokenizer(
if tokenizer_type: if tokenizer_type:
tokenizer = getattr(transformers, tokenizer_type).from_pretrained( tokenizer = getattr(transformers, tokenizer_type).from_pretrained(
base_model_config, 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: else:
tokenizer = AutoTokenizer.from_pretrained( tokenizer = AutoTokenizer.from_pretrained(