FitCheck é uma nova ferramenta que estima os requisitos máximos de VRAM para LoRA, QLoRA e ajuste fino completo de grandes modelos de linguagem sem exigir PyTorch, CUDA ou hardware GPU real.

  • Ele analisa a config.json do modelo no Hugging Face e metadados de contagem de parâmetros para relatar decomposição de memória, capacidade utilizável, margem e vereditos de ajuste.
  • A ferramenta suporta estimativas de serviço com base nos pesos do modelo e no cache KV, junto com um consultor para explorar tamanho do lote, comprimento da sequência e rank do LoRA.
  • Testes em uma Tesla T4 mostraram um erro absoluto médio de 2,4% através de 57 execuções de calibração e produziram 12/12 vereditos corretos de limite de ajuste em um teste de retenção separado.

A ferramenta ajuda os usuários a determinar se sua configuração de LML caberá na memória GPU antes de iniciar um trabalho, visando reduzir palpites e erros de falta de memória.