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
- Experimentos reproduzíveis
- Configurações complexas
- Compartilhar configurações com a equipe
Backend
Especifique onde o treinamento será executado:
aitraining llm --train --backend local ...
| Backend | Descrição |
|---|
local | Executar 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
| Backend | GPU |
|---|
spaces-t4-small | T4 (small) |
spaces-t4-medium | T4 (medium) |
spaces-a10g-small | A10G (small) |
spaces-a10g-large | A10G (large) |
spaces-a10g-largex2 | 2x A10G |
spaces-a10g-largex4 | 4x A10G |
spaces-a100-large | A100 |
spaces-l4x1 | 1x L4 |
spaces-l4x4 | 4x L4 |
spaces-l40sx1 | 1x L40S |
spaces-l40sx4 | 4x L40S |
spaces-l40sx8 | 8x L40S |
spaces-cpu-basic | Apenas CPU |
spaces-cpu-upgrade | CPU (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
