Arrakis / Modelos / Llama-2-7b-hf

Requisitos de GPU de Llama-2-7b-hf

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

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ónMemoria y configuración mínima
16 bits16 GB: una RTX 4090 o RTX 3090, la RTX 3090 desde 0,19 US$ por GPU-hora
8 bits8,1 GB: una RTX 4090 o RTX 3090, la RTX 3090 desde 0,19 US$ por GPU-hora
4 bits4,0 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étodoMemoria y configuración mínima
QLoRA4,7 GB: una RTX 4090 o RTX 3090, la RTX 3090 desde 0,19 US$ por GPU-hora
LoRA16 GB: una RTX 4090 o RTX 3090, la RTX 3090 desde 0,19 US$ por GPU-hora
Ajuste fino completo108 GB: una H200, desde 4,03 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
6.7B
Publicado en
BF16
Licencia
llama2
Fecha de publicación
2023-07-13
Pesos
meta-llama/Llama-2-7b-hf

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 meta-llama/Llama-2-7b-hf --tensor-parallel-size 1

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 Llama-2-7b-hf?
Para servirlo, unos 16 GB en 16 bits, 8 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, 16 GB con LoRA y 108 GB para un ajuste fino completo con Adam, sin contar las activaciones.
¿Puedo ejecutar Llama-2-7b-hf en una sola GPU?
Sí. En 16 bits necesita unos 16 GB, que caben en una RTX 4090 o RTX 3090.
¿Cuánto cuesta ejecutar Llama-2-7b-hf?
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 Llama-2-7b-hf en Arrakis?
Abre Models en la consola de Arrakis, pega meta-llama/Llama-2-7b-hf, 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.