Upload train_aviation.py with huggingface_hub

train_aviation.py

@@ -14,13 +14,34 @@
 
 import trackio
 import torch
+import os
+from huggingface_hub import list_repo_files
+
+# DEBUG: Check token and repo access
+print("🔍 DIAGNOSTICS:")
+token = os.environ.get("HF_TOKEN")
+print(f"   HF_TOKEN env var present: {bool(token)}")
+if token:
+    print(f"   HF_TOKEN prefix: {token[:4]}...")
+
+model_id = "mistralai/Ministral-3-14B-Reasoning-2512"
+try:
+    print(f"   Attempting to list files for {model_id}...")
+    files = list_repo_files(model_id, token=token)
+    print(f"   ✅ Success! Found {len(files)} files.")
+    print(f"   First 5 files: {files[:5]}")
+except Exception as e:
+    print(f"   ❌ Failed to list repo files: {e}")
+print("="*40)
+
 from datasets import load_dataset
 from peft import LoraConfig, prepare_model_for_kbit_training, get_peft_model
 from trl import SFTTrainer, SFTConfig
 from transformers import AutoModelForCausalLM, AutoTokenizer, BitsAndBytesConfig
 
 # Model ID
-model_id = "mistralai/Mistral-3-14B-Reasoning-2512"
+# model_id defined above
+
 
 # Load dataset
 print("📦 Loading dataset...")