Upload folder using huggingface_hub

src/train_hf.py

@@ -58,7 +58,10 @@ def train():
         fp16=False,                        # Enable if you have Tensor Cores (NVIDIA)
         gradient_checkpointing=True,       # CRITICAL for 5B model memory
         dataloader_num_workers=0,
-        report_to="tensorboard"
+        report_to="tensorboard",
+        push_to_hub=True,                  # Save artifacts to HF Hub
+        hub_model_id="Hexa09/hexa-tts-5b", # Target repo
+        hub_strategy="every_save"
     )
     
     trainer = Trainer(