Arrakis / Modèles / Llama-2-7b-hf

Besoins GPU de Llama-2-7b-hf

Llama-2-7b-hf compte 6,7B paramètres. Pour le servir, prévoyez environ 16 Go de mémoire GPU en 16 bits, 8,1 Go en 8 bits ou 4,0 Go en 4 bits, dont environ 20 % pour le cache KV et les activations.

Déployer ce modèle Vérifier un autre modèle

Ce qu’il faut pour le servir

La mémoire GPU pour le faire tourner en inférence, avec la plus petite configuration GPU qui le contient et le prix le plus bas sur Arrakis en ce moment.

Précision	Mémoire et plus petite configuration
16 bits	16 Go : un RTX 4090 ou RTX 3090, la RTX 3090 à partir de 0,19 $ par heure de GPU
8 bits	8,1 Go : un RTX 4090 ou RTX 3090, la RTX 3090 à partir de 0,19 $ par heure de GPU
4 bits	4,0 Go : un RTX 4090 ou RTX 3090, la RTX 3090 à partir de 0,19 $ par heure de GPU

Ce qu’il faut pour l’affiner

QLoRA entraîne des adaptateurs sur un modèle figé en 4 bits, LoRA sur un modèle en 16 bits, et le fine-tuning complet met à jour tous les poids. Les chiffres s’entendent hors activations, qui augmentent avec la taille des lots et la longueur des séquences.

Méthode	Mémoire et plus petite configuration
QLoRA	4,7 Go : un RTX 4090 ou RTX 3090, la RTX 3090 à partir de 0,19 $ par heure de GPU
LoRA	16 Go : un RTX 4090 ou RTX 3090, la RTX 3090 à partir de 0,19 $ par heure de GPU
Fine-tuning complet	108 Go : un H200, à partir de 3,44 $ par heure de GPU

À propos du modèle

D’après sa fiche et ses fichiers sur Hugging Face. Les chiffres de mémoire suivent le guide de la mémoire GPU.

Paramètres: 6.7B
Publié en: BF16
Licence: llama2
Date de publication: 2023-07-13
Poids: meta-llama/Llama-2-7b-hf

Déployez-le sur Arrakis

Ouvrez Models dans la console, choisissez vLLM, SGLang ou llama.cpp, et louez un GPU sur lequel il tient. Vous obtenez un endpoint compatible OpenAI avec sa propre clé d’API, et un accès SSH root à la machine. Ou faites-le tourner vous-même sur n’importe quelle machine avec assez de mémoire GPU :

vLLM

vllm serve meta-llama/Llama-2-7b-hf --tensor-parallel-size 1

Déployer ce modèle Comment fonctionne le déploiement

Questions

Les prix en temps réel de tous les GPU sont sur la page Prix des GPU.

De combien de mémoire GPU Llama-2-7b-hf a-t-il besoin ?

Pour le servir, environ 16 Go en 16 bits, 8 Go en 8 bits et 4 Go en 4 bits, dont environ 20 % pour le cache KV et les activations. Pour l’affiner, environ 5 Go avec QLoRA, 16 Go avec LoRA et 108 Go pour un fine-tuning complet avec Adam, hors activations.

Puis-je faire tourner Llama-2-7b-hf sur un seul GPU ?

Oui. En 16 bits, il lui faut environ 16 Go, ce qui tient sur un RTX 4090 ou RTX 3090.

Combien coûte l’exécution de Llama-2-7b-hf ?

En 16 bits, il lui faut un RTX 4090 ou RTX 3090 ; en 4 bits, un RTX 4090 ou RTX 3090. Chaque configuration de cette page affiche le prix le plus bas par heure de GPU sur Arrakis en ce moment, et vous payez à l’heure depuis un solde prépayé.

Comment déployer Llama-2-7b-hf sur Arrakis ?

Ouvrez Models dans la console Arrakis, collez meta-llama/Llama-2-7b-hf, choisissez vLLM, SGLang ou llama.cpp, et louez un GPU sur lequel il tient. Vous obtenez un endpoint compatible OpenAI avec sa propre clé d’API, et un accès SSH root à la machine.

Plus de modèles

Tous les modèles, et une vérification pour n’importe quel autre.