Saltar al contenido principal

​
Opciones Globales

Estas opciones funcionan en la mayoría de comandos del CLI de AITraining.

​
Versión

Verifica la versión instalada: 
aitraining --version
aitraining -v

​
Ayuda

Obtén ayuda para cualquier comando: 
aitraining --help
aitraining llm --help
aitraining text-classification --help

​
Archivo de Configuración

Carga parámetros desde un archivo de configuración YAML: 
aitraining --config path/to/config.yaml
Esto es útil para:
  • Experimentos reproducibles
  • Configuraciones complejas
  • Compartir configuraciones con compañeros de equipo

​
Backend

Especifica dónde se ejecuta el entrenamiento: 
aitraining llm --train --backend local ...
Backends disponibles:
BackendDescripción
localEjecutar en máquina local (predeterminado). Variantes: local-cli, local-ui
spaces-*Ejecutar en Hugging Face Spaces
ep-*Hugging Face Endpoints
ngc-*NVIDIA NGC/DGX Cloud
nvcf-*NVIDIA Cloud Functions

​
Opciones de Backend Spaces

BackendGPU
spaces-t4-smallT4 (small)
spaces-t4-mediumT4 (medium)
spaces-a10g-smallA10G (small)
spaces-a10g-largeA10G (large)
spaces-a10g-largex22x A10G
spaces-a10g-largex44x A10G
spaces-a100-largeA100
spaces-l4x11x L4
spaces-l4x44x L4
spaces-l40sx11x L40S
spaces-l40sx44x L40S
spaces-l40sx88x L40S
spaces-cpu-basicSolo CPU
spaces-cpu-upgradeCPU (mejorado)
Los backends remotos requieren autenticación: Al usar backends no locales (spaces-*, ep-*, ngc-*, nvcf-*), debes proporcionar --username y --token para autenticación en Hugging Face.
Push a Hub también requiere autenticación: Incluso con --backend local, usar --push-to-hub requiere --username y --token para subir el modelo a Hugging Face Hub.

​
Variables de Entorno

Configura estas antes de ejecutar comandos:

​
Autenticación

export HF_TOKEN="hf_..."        # Token de Hugging Face
export WANDB_API_KEY="..."       # Clave de Weights & Biases

​
Configuración de GPU

export CUDA_VISIBLE_DEVICES=0,1  # Usar GPUs específicas
export PYTORCH_CUDA_ALLOC_CONF=max_split_size_mb:512  # Gestión de memoria

​
Control MPS (Apple Silicon)

export AUTOTRAIN_DISABLE_MPS=1  # Forzar entrenamiento CPU en Mac
export AUTOTRAIN_ENABLE_MPS=1   # Forzar MPS incluso con cuantización

​
Modo Interactivo

Lanza el asistente de configuración: 
aitraining           # Sin argumentos = modo asistente
aitraining llm --interactive  # Modo interactivo explícito

​
Próximos Pasos

Configuraciones YAML

Formato de archivo de configuración

Estructura de Comandos

Referencia completa de comandos