Skip to main content

Interactive Wizard

The AITraining Interactive Wizard guides you step-by-step through training your first AI model. No configuration files, no command memorization - just answer questions and start training.

Why Use the Wizard?

Zero Configuration

Answer simple questions instead of writing YAML files or memorizing CLI flags.

Smart Defaults

Intelligent defaults based on your hardware, model choice, and dataset size.

Validation Built-in

Catches errors before training starts - invalid datasets, missing columns, incompatible settings.

Learn As You Go

Every step has :help - learn what each option does while configuring.

Launch the Wizard

aitraining
That’s it. Just run aitraining with no arguments and the wizard starts.

What You’ll See

 █████╗ ██╗████████╗██████╗  █████╗ ██╗███╗   ██╗██╗███╗   ██╗ ██████╗
██╔══██╗██║╚══██╔══╝██╔══██╗██╔══██╗██║████╗  ██║██║████╗  ██║██╔════╝
███████║██║   ██║   ██████╔╝███████║██║██╔██╗ ██║██║██╔██╗ ██║██║  ███╗
██╔══██║██║   ██║   ██╔══██╗██╔══██║██║██║╚██╗██║██║██║╚██╗██║██║   ██║
██║  ██║██║   ██║   ██║  ██║██║  ██║██║██║ ╚████║██║██║ ╚████║╚██████╔╝
╚═╝  ╚═╝╚═╝   ╚═╝   ╚═╝  ╚═╝╚═╝  ╚═╝╚═╝╚═╝  ╚═══╝╚═╝╚═╝  ╚═══╝ ╚═════╝
        From zero to hero Machine Learning Training Platform

🚀 Welcome to the AITraining Interactive Setup!

I'll help you configure your training job step by step.
Press Enter to accept defaults shown in [brackets].

Commands available at any prompt:
  :back  → return to the previous step
  :help  → show extra information for the current prompt
  :exit  → cancel setup

Wizard Commands

These commands work at any prompt:
CommandWhat it Does
:backGo back to the previous step
:helpShow detailed help for the current question
:exitCancel the wizard
/search <query>Search for models or datasets
/sortChange how models/datasets are sorted
/filterFilter models by size (small/medium/large)
/refreshReload the model/dataset list

Wizard Steps

The wizard guides you through these steps:
  1. Trainer Type - What kind of AI? (LLM, classification, vision, etc.)
  2. Training Method - For LLMs: SFT, DPO, ORPO, or PPO
  3. Project Name - Where to save your trained model
  4. Model Selection - Choose a base model (with hardware recommendations)
  5. Dataset Configuration - Select and configure your training data
  6. Advanced Options - Optional: hyperparameters, LoRA, logging

Quick Start Paths

  • First Timer
  • Have Your Data
  • Specific Model
Just want to try it out?
  1. Run aitraining
  2. Select “LLM” (option 1)
  3. Select “SFT” (option 1)
  4. Accept all defaults
  5. Watch it train!
The wizard will pick a small model and sample dataset automatically.

What’s Next?