From dab2590e4d428e849c7cc35bf1eaf926ddf31ca3 Mon Sep 17 00:00:00 2001 From: NanoCode012 Date: Thu, 10 Oct 2024 18:07:00 +0700 Subject: [PATCH] chore: refactor --- src/axolotl/utils/chat_templates.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/axolotl/utils/chat_templates.py b/src/axolotl/utils/chat_templates.py index 43d0f6d5d..f3dc8f5fc 100644 --- a/src/axolotl/utils/chat_templates.py +++ b/src/axolotl/utils/chat_templates.py @@ -90,10 +90,10 @@ def get_chat_template( def extract_chat_template_args(cfg, ds_cfg: Optional[Dict[str, Any]] = None): if ds_cfg and ds_cfg.get("chat_template"): - chat_template_choice = ds_cfg.get("chat_template") or "tokenizer_default" + chat_template_choice = ds_cfg.get("chat_template") or _DEFAULT_TEMPLATE_CHOICE chat_template_jinja = ds_cfg.get("chat_template_jinja") else: - chat_template_choice = cfg.get("chat_template") or "tokenizer_default" + chat_template_choice = cfg.get("chat_template") or _DEFAULT_TEMPLATE_CHOICE chat_template_jinja = cfg.get("chat_template_jinja") return chat_template_choice, chat_template_jinja