fix run name for wandb and defaults for chat template fields

src/axolotl/core/trainer_builder.py

@@ -1417,6 +1417,8 @@ def build(self, total_num_steps):
  report_to = []
  if self.cfg.use_wandb:
  report_to.append("wandb")
+ if self.cfg.wandb_name:
+ training_arguments_kwargs["run_name"] = self.cfg.wandb_name
  if self.cfg.use_mlflow:
  report_to.append("mlflow")
  if self.cfg.use_tensorboard:
@@ -1574,6 +1576,12 @@ def build(self, total_num_steps):
  )
  training_args = self.hook_post_create_training_args(training_args)
 
+ # unset run_name so wandb sets up experiment names
+ if self.cfg.use_wandb and training_args.run_name == training_args.output_dir:
+ training_args.run_name = ( # pylint: disable=attribute-defined-outside-init
+ None
+ )
+
  data_collator_kwargs = {
  "padding": True, # True/"longest" is the default
  }

src/axolotl/prompt_strategies/chat_template.py

@@ -375,8 +375,8 @@ 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")),
- "message_field_role": ds_cfg.get("message_field_role", "from"),
- "message_field_content": ds_cfg.get("message_field_content", "value"),
+ "message_field_role": ds_cfg.get("message_field_role", "role"),
+ "message_field_content": ds_cfg.get("message_field_content", "content"),
  "message_field_training": ds_cfg.get("message_field_training", None),
  "message_field_training_detail": ds_cfg.get(
  "message_field_training_detail",