Upload train_aviation.py with huggingface_hub

train_aviation.py

@@ -18,26 +18,32 @@ 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)
+# ... (diagnostics code kept as is) ...
 
 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
+from transformers import AutoModelForCausalLM, AutoTokenizer, BitsAndBytesConfig, AutoConfig
+
+# Register 'ministral3' config if needed
+print("🔧 Registering ministral3 config...")
+try:
+    # Try to find Mistral3Config (available in very recent transformers)
+    from transformers import Mistral3Config
+    AutoConfig.register("ministral3", Mistral3Config)
+    print("   Registered ministral3 -> Mistral3Config")
+except ImportError:
+    try:
+        # Fallback to internal path if not exposed
+        from transformers.models.mistral3.configuration_mistral3 import Mistral3Config
+        AutoConfig.register("ministral3", Mistral3Config)
+        print("   Registered ministral3 -> Mistral3Config (internal)")
+    except ImportError:
+        # Fallback to standard MistralConfig
+        print("   Mistral3Config not found, falling back to MistralConfig")
+        from transformers import MistralConfig
+        AutoConfig.register("ministral3", MistralConfig)
+        print("   Registered ministral3 -> MistralConfig")
 
 # Model ID
 # model_id defined above