Upload train_aviation.py with huggingface_hub

train_aviation.py

@@ -42,17 +42,18 @@ from transformers import AutoModelForCausalLM, AutoTokenizer, BitsAndBytesConfig
 # Register 'ministral3' config to handle nested text_config
 print("🔧 Registering ministral3 config...")
 try:
-    from transformers import MinistralConfig, MinistralModel, AutoModel
-    class Ministral3CompatConfig(MinistralConfig):
+    # Use standard MistralConfig (not MinistralConfig) as base
+    from transformers import MistralConfig, MistralModel, AutoModel, AutoConfig
+    class Ministral3CompatConfig(MistralConfig):
         model_type = "ministral3"
     
     # 1. Register Config
     AutoConfig.register("ministral3", Ministral3CompatConfig)
-    print("   Registered ministral3 -> Ministral3CompatConfig (subclass)")
+    print("   Registered ministral3 -> Ministral3CompatConfig (subclass of MistralConfig)")
 
-    # 2. Register Model for this config (for AutoModel.from_config inside Mistral3Model)
-    AutoModel.register(Ministral3CompatConfig, MinistralModel)
-    print("   Registered Ministral3CompatConfig -> MinistralModel")
+    # 2. Register Model for this config
+    AutoModel.register(Ministral3CompatConfig, MistralModel)
+    print("   Registered Ministral3CompatConfig -> MistralModel")
 
 except Exception as e:
     print(f"   ❌ Failed to register ministral3 config/model: {e}")
@@ -60,6 +61,7 @@ except Exception as e:
 # Register Mistral3Config to a model class
 # ... (rest of registration kept as is)
 # ... (rest of registration kept as is)
+# ... (rest of registration kept as is)
 print("🔧 Registering Mistral3 model class...")
 try:
     from transformers.models.mistral3.configuration_mistral3 import Mistral3Config