fix Trackio crash: push at end only

train_dpo.py

@@ -93,18 +93,19 @@ def main() -> None:
             logging_steps=5,
             eval_strategy="steps",
             eval_steps=20,
+            save_strategy="epoch",
+            save_total_limit=1,
             max_length=1024,
             max_prompt_length=512,
             bf16=True,
-            push_to_hub=True,
+            push_to_hub=False,
             hub_model_id=HUB_MODEL_ID,
-            report_to="trackio",
-            project="cpp-compiler-rl",
-            run_name="dpo-compiler-prefs",
+            report_to="none",
         ),
     )
     trainer.train()
-    trainer.push_to_hub()
+    trainer.model.push_to_hub(HUB_MODEL_ID, private=False)
+    tokenizer.push_to_hub(HUB_MODEL_ID, private=False)
     print(f"Pushed to {HUB_MODEL_ID}")