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

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

Mistral-7B-v0.1 tiene 7,2B parámetros. Para servirlo, calcula unos 17 GB de memoria de GPU en 16 bits, 8,7 GB en 8 bits o 4,3 GB en 4 bits, incluido un 20% para la caché KV y las activaciones.

Desplegar este modelo Comprobar otro modelo

Lo que hace falta para servirlo

La memoria de GPU para ejecutarlo en inferencia, con la configuración de GPU más pequeña que lo sostiene y el precio más bajo en Arrakis ahora mismo.

Precisión	Memoria y configuración mínima
16 bits	17 GB: una RTX 4090 o RTX 3090, la RTX 3090 desde 0,19 US$ por GPU-hora
8 bits	8,7 GB: una RTX 4090 o RTX 3090, la RTX 3090 desde 0,19 US$ por GPU-hora
4 bits	4,3 GB: una RTX 4090 o RTX 3090, la RTX 3090 desde 0,19 US$ por GPU-hora

Lo que hace falta para ajustarlo

QLoRA entrena adaptadores sobre un modelo congelado de 4 bits; LoRA, sobre uno de 16 bits, y el ajuste fino completo actualiza todos los pesos. Las cifras no incluyen las activaciones, que crecen con el tamaño del lote y la longitud de secuencia.

Método	Memoria y configuración mínima
QLoRA	5,1 GB: una RTX 4090 o RTX 3090, la RTX 3090 desde 0,19 US$ por GPU-hora
LoRA	17 GB: una RTX 4090 o RTX 3090, la RTX 3090 desde 0,19 US$ por GPU-hora
Ajuste fino completo	116 GB: una H200, desde 3,44 US$ por GPU-hora

Sobre el modelo

De su ficha y sus archivos en Hugging Face. Las cifras de memoria siguen la guía de memoria de GPU.

Parámetros: 7.2B
Ventana de contexto (tokens): 32,768
Publicado en: BF16
Licencia: apache-2.0
Fecha de publicación: 2023-09-20
Pesos: mistralai/Mistral-7B-v0.1

Despliégalo en Arrakis

Abre Models en la consola, elige vLLM, SGLang o llama.cpp y alquila una GPU en la que quepa. Obtienes un endpoint compatible con OpenAI con su propia clave de API, y SSH root a la máquina. O ejecútalo tú mismo en cualquier máquina con suficiente memoria de GPU:

vLLM

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

Desplegar este modelo Cómo funciona el despliegue

Preguntas

Los precios en tiempo real de todas las GPU están en la página de precios de GPU.

¿Cuánta memoria de GPU necesita Mistral-7B-v0.1?

Para servirlo, unos 17 GB en 16 bits, 9 GB en 8 bits y 4 GB en 4 bits, incluido un 20% para la caché KV y las activaciones. Para ajustarlo, unos 5 GB con QLoRA, 17 GB con LoRA y 116 GB para un ajuste fino completo con Adam, sin contar las activaciones.

¿Puedo ejecutar Mistral-7B-v0.1 en una sola GPU?

Sí. En 16 bits necesita unos 17 GB, que caben en una RTX 4090 o RTX 3090.

¿Cuánto cuesta ejecutar Mistral-7B-v0.1?

En 16 bits necesita una RTX 4090 o RTX 3090; en 4 bits, una RTX 4090 o RTX 3090. Cada configuración de esta página muestra el precio más bajo por GPU-hora en Arrakis ahora mismo, y pagas por hora desde un saldo prepago.

¿Cómo despliego Mistral-7B-v0.1 en Arrakis?

Abre Models en la consola de Arrakis, pega mistralai/Mistral-7B-v0.1, elige vLLM, SGLang o llama.cpp y alquila una GPU en la que quepa. Obtienes un endpoint compatible con OpenAI con su propia clave de API, y SSH root a la máquina.

Más modelos

Todos los modelos, y una comprobación para cualquier otro.