Fix: add hub_token for push_to_hub

train.py

@@ -192,6 +192,10 @@ class D1337CipherTrainer:
             
             # Training arguments (standard TrainingArguments)
             self.training_status = "Setting up training..."
+            
+            # Get HF token from environment
+            hf_token = os.environ.get("HF_TOKEN", None)
+            
             training_args = TrainingArguments(
                 output_dir="./d1337-cipher-output",
                 num_train_epochs=self.config.num_epochs,
@@ -211,9 +215,10 @@ class D1337CipherTrainer:
                 group_by_length=True,
                 dataloader_num_workers=4,
                 remove_unused_columns=False,
-                push_to_hub=True,
+                push_to_hub=True if hf_token else False,
                 hub_model_id=self.config.output_model,
                 hub_private_repo=True,
+                hub_token=hf_token,
                 report_to="none",
             )