Pular para o conteúdo principal

​
Opções Globais

Essas opções funcionam na maioria dos comandos do AITraining CLI.

​
Versão

Verifique a versão instalada: 
aitraining --version
aitraining -v

​
Ajuda

Obtenha ajuda para qualquer comando: 
aitraining --help
aitraining llm --help
aitraining text-classification --help

​
Arquivo de Configuração

Carregue parâmetros de um arquivo de configuração YAML: 
aitraining --config path/to/config.yaml
Isso é útil para:
  • Experimentos reproduzíveis
  • Configurações complexas
  • Compartilhar configurações com a equipe

​
Backend

Especifique onde o treinamento será executado: 
aitraining llm --train --backend local ...
Backends disponíveis:
BackendDescrição
localExecutar na máquina local (padrão). Variantes: local-cli, local-ui
spaces-*Executar no Hugging Face Spaces
ep-*Hugging Face Endpoints
ngc-*NVIDIA NGC/DGX Cloud
nvcf-*NVIDIA Cloud Functions

​
Opções 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-basicApenas CPU
spaces-cpu-upgradeCPU (atualizado)
Backends remotos requerem autenticação: Ao usar backends não locais (spaces-*, ep-*, ngc-*, nvcf-*), você deve fornecer --username e --token para autenticação no Hugging Face.
Push para Hub também requer autenticação: Mesmo com --backend local, usar --push-to-hub requer --username e --token para fazer upload do modelo para o Hugging Face Hub.

​
Variáveis de Ambiente

Configure essas variáveis antes de executar comandos:

​
Autenticação

export HF_TOKEN="hf_..."        # Token do Hugging Face
export WANDB_API_KEY="..."       # Chave do Weights & Biases

​
Configuração de GPU

export CUDA_VISIBLE_DEVICES=0,1  # Usar GPUs específicas
export PYTORCH_CUDA_ALLOC_CONF=max_split_size_mb:512  # Gerenciamento de memória

​
Controle MPS (Apple Silicon)

export AUTOTRAIN_DISABLE_MPS=1  # Forçar treinamento CPU no Mac
export AUTOTRAIN_ENABLE_MPS=1   # Forçar MPS mesmo com quantização

​
Modo Interativo

Inicie o assistente de configuração: 
aitraining           # Sem argumentos = modo assistente
aitraining llm --interactive  # Modo interativo explícito

​
Próximos Passos

Configurações YAML

Formato de arquivo de configuração

Estrutura de Comandos

Referência completa de comandos