Upload train_aviation.py with huggingface_hub

train_aviation.py

@@ -25,25 +25,17 @@ 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 if needed
+# Register 'ministral3' config to handle nested text_config
 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")
+    from transformers import MistralConfig
+    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}")
 
 # Model ID
 # model_id defined above