# PyPilot Training Script
import torch
from modeling_pypilot import PyPilotModel, PyPilotConfig

def train_pypilot():
    print("🚀 Starting PyPilot Training...")
    
    # Model configuration
    config = PyPilotConfig()
    model = PyPilotModel(config)
    
    print("✅ PyPilot model created successfully!")
    print(f"📊 Model parameters: {sum(p.numel() for p in model.parameters()):,}")
    
    # Training setup (simplified for now)
    optimizer = torch.optim.AdamW(model.parameters(), lr=0.001)
    
    print("🎯 Training ready to begin!")
    return model

if __name__ == "__main__":
    train_pypilot()