From 78b42a3fe13c49e317bc116b9999c30e070322cc Mon Sep 17 00:00:00 2001 From: Wing Lian Date: Tue, 30 Jul 2024 20:58:17 -0400 Subject: [PATCH] fix roles to train defaults and make logging less verbose (#1801) --- src/axolotl/prompt_strategies/chat_template.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/axolotl/prompt_strategies/chat_template.py b/src/axolotl/prompt_strategies/chat_template.py index f9fa71f21..719d79505 100644 --- a/src/axolotl/prompt_strategies/chat_template.py +++ b/src/axolotl/prompt_strategies/chat_template.py @@ -10,8 +10,8 @@ from axolotl.prompters import IGNORE_TOKEN_ID, Prompter from axolotl.utils.chat_templates import chat_templates # Configure the logger -logging.basicConfig(level=logging.DEBUG) LOG = logging.getLogger("axolotl") +LOG.setLevel(logging.INFO) class ChatTemplatePrompter(Prompter): @@ -355,7 +355,7 @@ def load(tokenizer, cfg, ds_cfg: Optional[Dict[str, Any]] = None): strategy_params = { "train_on_inputs": cfg.train_on_inputs, "sequence_len": cfg.sequence_len, - "roles_to_train": ds_cfg.get("roles_to_train"), + "roles_to_train": ds_cfg.get("roles_to_train", ["gpt", "assistant"]), "train_on_eos": ds_cfg.get("train_on_eos", "last"), }