From 4aafb7e600cc847f20d5c4531a5063238023e75b Mon Sep 17 00:00:00 2001 From: NanoCode012 Date: Mon, 14 Oct 2024 23:41:17 +0700 Subject: [PATCH] fix: imported name incorrectly updated on merge --- src/axolotl/prompt_strategies/bradley_terry/chat_template.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/axolotl/prompt_strategies/bradley_terry/chat_template.py b/src/axolotl/prompt_strategies/bradley_terry/chat_template.py index ccda0a4bd..d3cd5c2f5 100644 --- a/src/axolotl/prompt_strategies/bradley_terry/chat_template.py +++ b/src/axolotl/prompt_strategies/bradley_terry/chat_template.py @@ -8,7 +8,7 @@ from axolotl.prompt_strategies.chat_template import ( ChatTemplatePrompter, ChatTemplateStrategy, ) -from axolotl.utils.chat_templates import chat_templates +from axolotl.utils.chat_templates import get_chat_template class BTChatTemplateStrategy(ChatTemplateStrategy): @@ -56,7 +56,7 @@ def load(tokenizer, cfg, ds_cfg: Optional[Dict[str, Any]] = None): prompter_params = { "tokenizer": tokenizer, - "chat_template": chat_templates(ds_cfg.get("chat_template", "chatml")), + "chat_template": get_chat_template(ds_cfg.get("chat_template", "chatml")), "message_field_role": ds_cfg.get("message_field_role", "from"), "message_field_content": ds_cfg.get("message_field_content", "value"), "message_field_training": ds_cfg.get("message_field_training", "training"),