From 31d23504a5e58ad2927014dccb7ee2a005a6a42c Mon Sep 17 00:00:00 2001
From: Hamel Husain <hamel.husain@gmail.com>
Date: Thu, 4 Jan 2024 18:13:54 -0800
Subject: [PATCH] fix model card upload for PEFT models (#1043)

---
 src/axolotl/train.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/src/axolotl/train.py b/src/axolotl/train.py
index e0da11252..cf3382c89 100644
--- a/src/axolotl/train.py
+++ b/src/axolotl/train.py
@@ -188,6 +188,9 @@ def train(
 
     if not cfg.hub_model_id:
         trainer.create_model_card(model_name=cfg.output_dir.lstrip("./"))
+    elif cfg.hub_model_id:
+        # defensively push to the hub to ensure the model card is updated
+        trainer.push_to_hub()
 
     return model, tokenizer