Skip to content

Commit

Permalink
fix roles to train defaults and make logging less verbose (#1801)
Browse files Browse the repository at this point in the history
  • Loading branch information
winglian authored Jul 31, 2024
1 parent 3ebf224 commit 78b42a3
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions src/axolotl/prompt_strategies/chat_template.py
Original file line number Diff line number Diff line change
Expand Up @@ -10,8 +10,8 @@
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):
Expand Down Expand Up @@ -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"),
}

Expand Down

0 comments on commit 78b42a3

Please sign in to comment.