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

fix: imported name incorrectly updated on merge

Browse Source
This commit is contained in:
NanoCode012
2024-10-14 23:41:17 +07:00
parent 17bc4c8b36
commit 4aafb7e600
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/axolotl/prompt_strategies/bradley_terry/chat_template.py
View File

@@ -8,7 +8,7 @@ from axolotl.prompt_strategies.chat_template import (
ChatTemplatePrompter, ChatTemplatePrompter,
ChatTemplateStrategy, ChatTemplateStrategy,
) )
from axolotl.utils.chat_templates import chat_templates from axolotl.utils.chat_templates import get_chat_template
class BTChatTemplateStrategy(ChatTemplateStrategy): class BTChatTemplateStrategy(ChatTemplateStrategy):
@@ -56,7 +56,7 @@ def load(tokenizer, cfg, ds_cfg: Optional[Dict[str, Any]] = None):
prompter_params = { prompter_params = {
"tokenizer": tokenizer, "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_role": ds_cfg.get("message_field_role", "from"),
"message_field_content": ds_cfg.get("message_field_content", "value"), "message_field_content": ds_cfg.get("message_field_content", "value"),
"message_field_training": ds_cfg.get("message_field_training", "training"), "message_field_training": ds_cfg.get("message_field_training", "training"),