Arrakis / Modelos / Mistral-7B-v0.1

Requisitos de GPU do Mistral-7B-v0.1

O Mistral-7B-v0.1 tem 7,2B parâmetros. Para servi-lo, conte com cerca de 17 GB de memória de GPU em 16 bits, 8,7 GB em 8 bits ou 4,3 GB em 4 bits, incluindo cerca de 20% para o cache KV e as ativações.

O que é preciso para servi-lo

A memória de GPU para rodá-lo em inferência, com a menor configuração de GPU que o comporta e o menor preço na Arrakis agora.

PrecisãoMemória e menor configuração
16 bits17 GB: uma RTX 4090 ou RTX 3090, a RTX 3090 a partir de US$ 0,19 por GPU-hora
8 bits8,7 GB: uma RTX 4090 ou RTX 3090, a RTX 3090 a partir de US$ 0,19 por GPU-hora
4 bits4,3 GB: uma RTX 4090 ou RTX 3090, a RTX 3090 a partir de US$ 0,19 por GPU-hora

O que é preciso para fazer ajuste fino

O QLoRA treina adaptadores sobre um modelo congelado em 4 bits, o LoRA sobre um em 16 bits, e o ajuste fino completo atualiza todos os pesos. Os valores não incluem as ativações, que crescem com o tamanho do lote e o comprimento da sequência.

MétodoMemória e menor configuração
QLoRA5,1 GB: uma RTX 4090 ou RTX 3090, a RTX 3090 a partir de US$ 0,19 por GPU-hora
LoRA17 GB: uma RTX 4090 ou RTX 3090, a RTX 3090 a partir de US$ 0,19 por GPU-hora
Ajuste fino completo116 GB: uma H200, a partir de US$ 3,44 por GPU-hora

Sobre o modelo

Do cartão e dos arquivos no Hugging Face. Os números de memória seguem o guia de memória de GPU.

Parâmetros
7.2B
Janela de contexto (tokens)
32,768
Publicado em
BF16
Licença
apache-2.0
Data de lançamento
2023-09-20
Pesos
mistralai/Mistral-7B-v0.1

Implante na Arrakis

Abra Models no console, escolha vLLM, SGLang ou llama.cpp e alugue uma GPU em que ele caiba. Você recebe um endpoint compatível com OpenAI com sua própria chave de API, e SSH root na máquina. Ou rode você mesmo em qualquer máquina com memória de GPU suficiente:

vLLM
vllm serve mistralai/Mistral-7B-v0.1 --tensor-parallel-size 1

Perguntas

Os preços em tempo real de todas as GPUs estão na página de preços de GPU.

Quanta memória de GPU o Mistral-7B-v0.1 precisa?
Para servi-lo, cerca de 17 GB em 16 bits, 9 GB em 8 bits e 4 GB em 4 bits, incluindo cerca de 20% para o cache KV e as ativações. Para fazer ajuste fino, cerca de 5 GB com QLoRA, 17 GB com LoRA e 116 GB para ajuste fino completo com Adam, sem contar as ativações.
Posso rodar o Mistral-7B-v0.1 em uma única GPU?
Sim. Em 16 bits ele precisa de cerca de 17 GB, que cabem em uma RTX 4090 ou RTX 3090.
Quanto custa rodar o Mistral-7B-v0.1?
Em 16 bits ele precisa de uma RTX 4090 ou RTX 3090; em 4 bits, de uma RTX 4090 ou RTX 3090. Cada configuração desta página mostra o menor preço por GPU-hora na Arrakis agora, e você paga por hora a partir de um saldo pré-pago.
Como implanto o Mistral-7B-v0.1 na Arrakis?
Abra Models no console da Arrakis, cole mistralai/Mistral-7B-v0.1, escolha vLLM, SGLang ou llama.cpp e alugue uma GPU em que ele caiba. Você recebe um endpoint compatível com OpenAI com sua própria chave de API, e SSH root na máquina.