From 37e9da7a534f8799725a9cfe6a410e91bbd7187a Mon Sep 17 00:00:00 2001
From: madScientist10 <42779409+madScientist10@users.noreply.github.com>
Date: Tue, 10 Feb 2026 12:53:09 +0200
Subject: [PATCH] add hub_revision support for specifying branch when pushing
 checkpoints (#3387) [skip ci]

---
 src/axolotl/core/builders/base.py  | 3 +++
 src/axolotl/utils/schemas/model.py | 6 ++++++
 2 files changed, 9 insertions(+)

diff --git a/src/axolotl/core/builders/base.py b/src/axolotl/core/builders/base.py
index f3a965435..c23433866 100644
--- a/src/axolotl/core/builders/base.py
+++ b/src/axolotl/core/builders/base.py
@@ -409,6 +409,9 @@ class TrainerBuilderBase(abc.ABC):
             if self.cfg.hub_strategy:
                 training_args_kwargs["hub_strategy"] = self.cfg.hub_strategy
 
+            if self.cfg.hub_revision:
+                training_args_kwargs["hub_revision"] = self.cfg.hub_revision
+
     def _configure_save_and_eval_strategy(self, training_args_kwargs: dict):
         # save_strategy and save_steps
         if self.cfg.save_steps:
diff --git a/src/axolotl/utils/schemas/model.py b/src/axolotl/utils/schemas/model.py
index 31de7b45e..02b971c1d 100644
--- a/src/axolotl/utils/schemas/model.py
+++ b/src/axolotl/utils/schemas/model.py
@@ -120,6 +120,12 @@ class ModelOutputConfig(BaseModel):
         default=None,
         json_schema_extra={"description": "how to push checkpoints to hub"},
     )
+    hub_revision: str | None = Field(
+        default=None,
+        json_schema_extra={
+            "description": "branch/revision to push to on hub (default: main)"
+        },
+    )
     save_safetensors: bool | None = Field(
         default=True,
         json_schema_extra={