aitraining --version
aitraining -v

aitraining --help
aitraining llm --help
aitraining text-classification --help

aitraining --config path/to/config.yaml

aitraining llm --train --backend local ...

export HF_TOKEN="hf_..."        # Hugging Face token
export WANDB_API_KEY="..."       # Weights & Biases key

export CUDA_VISIBLE_DEVICES=0,1  # 使用特定 GPU
export PYTORCH_CUDA_ALLOC_CONF=max_split_size_mb:512  # 内存管理

export AUTOTRAIN_DISABLE_MPS=1  # 在 Mac 上强制 CPU 训练
export AUTOTRAIN_ENABLE_MPS=1   # 即使使用量化也强制 MPS

aitraining           # 无参数 = 向导模式
aitraining llm --interactive  # 显式交互模式