Upload train_aviation.py with huggingface_hub

train_aviation.py

@@ -39,6 +39,18 @@ from peft import LoraConfig, prepare_model_for_kbit_training, get_peft_model
 from trl import SFTTrainer, SFTConfig
 from transformers import AutoModelForCausalLM, AutoTokenizer, BitsAndBytesConfig, AutoConfig
 
+# Register 'ministral3' config to handle nested text_config
+print("🔧 Registering ministral3 config...")
+try:
+    from transformers import MistralConfig, AutoConfig
+    class MinistralConfig(MistralConfig):
+        model_type = "ministral3"
+    
+    AutoConfig.register("ministral3", MinistralConfig)
+    print("   Registered ministral3 -> MinistralConfig (subclass)")
+except Exception as e:
+    print(f"   ❌ Failed to register ministral3 config: {e}")
+
 # Register Mistral3Config to a model class
 print("🔧 Registering Mistral3 model class...")
 try: