diff --git a/donut-train.py b/donut-train.py
index ff4c3de..8eaa167 100644
--- a/donut-train.py
+++ b/donut-train.py
@@ -328,7 +328,7 @@ class PushToHubCallback(Callback):
         print(f"Pushing model to the hub, epoch {trainer.current_epoch}")
         pl_module.model.push_to_hub(OUTPUT_MODEL_PATH,
                                     commit_message=f"Training in progress, epoch {trainer.current_epoch}")
-        pl_module.processor.push_to_hub(OUTPUT_MODEL_PATH, commit_message=f"Training in progress, epoch {trainer.current_epoch}")
+        # pl_module.processor.push_to_hub(OUTPUT_MODEL_PATH, commit_message=f"Training in progress, epoch {trainer.current_epoch}")
 
     def on_train_end(self, trainer, pl_module):
         print(f"Pushing model to the hub after training")