Saltar al contenido principal

Documentation Index

Fetch the complete documentation index at: https://docs.monostate.ai/llms.txt

Use this file to discover all available pages before exploring further.

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