diff --git a/test2.yml b/test2.yml
index b5fa3d39e..d8f861e32 100644
--- a/test2.yml
+++ b/test2.yml
@@ -29,4 +29,10 @@ gradient_checkpointing: True
 gradient_checkpointing_kwargs:
   use_reentrant: false
 #special_tokens:
-#  pad_token: <|end_of_text|>
\ No newline at end of file
+#  pad_token: <|end_of_text|>
+
+wandb_project: check_dpotrainer
+wandb_entity: axolotl-ai
+wandb_watch:
+wandb_name: baseline/dpo_base/dpo_use_weighting
+wandb_log_model:
\ No newline at end of file