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

chore: refactor

Browse Source
This commit is contained in:
NanoCode012
2024-10-10 18:07:00 +07:00
parent e5162b7a41
commit dab2590e4d
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/axolotl/utils/chat_templates.py
View File

@@ -90,10 +90,10 @@ def get_chat_template(
def extract_chat_template_args(cfg, ds_cfg: Optional[Dict[str, Any]] = None): def extract_chat_template_args(cfg, ds_cfg: Optional[Dict[str, Any]] = None):
if ds_cfg and ds_cfg.get("chat_template"): 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") chat_template_jinja = ds_cfg.get("chat_template_jinja")
else: 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") chat_template_jinja = cfg.get("chat_template_jinja")
return chat_template_choice, chat_template_jinja return chat_template_choice, chat_template_jinja