Upload train_aviation.py with huggingface_hub

train_aviation.py

@@ -33,6 +33,24 @@ class RegistrableMinistralConfig(MistralConfig):
 AutoConfig.register("ministral3", RegistrableMinistralConfig)
 print("🔧 Registered 'ministral3' to RegistrableMinistralConfig.")
 
+# Register Mistral3Config to a model class
+print("🔧 Registering Mistral3 model class with AutoModelForCausalLM...")
+try:
+    from transformers.models.mistral3.configuration_mistral3 import Mistral3Config
+    from transformers.models.mistral3.modeling_mistral3 import Mistral3ForConditionalGeneration
+    AutoModelForCausalLM.register(Mistral3Config, Mistral3ForConditionalGeneration)
+    print("   Registered Mistral3Config -> Mistral3ForConditionalGeneration")
+except ImportError as e:
+    print(f"   ❌ Failed to import Mistral3 modeling classes: {e}")
+    # Fallback if specific Mistral3 classes are not directly importable
+    print("   Trying to register Mistral3Config to standard MistralForCausalLM as fallback.")
+    from transformers import MistralForCausalLM
+    try:
+        AutoModelForCausalLM.register(Mistral3Config, MistralForCausalLM)
+        print("   Registered Mistral3Config -> MistralForCausalLM (fallback)")
+    except Exception as fallback_e:
+        print(f"   ❌ Fallback registration also failed: {fallback_e}")
+
 
 # Load dataset
 print("📦 Loading dataset...")
@@ -160,4 +178,4 @@ print("🚀 Starting training...")
 trainer.train()
 
 print("💾 Pushing to Hub...")
-trainer.push_to_hub()
+trainer.push_to_hub()