Transformers documentation
SpQR
Commencer
Classes de base
Modèles
Préprocesseurs
Inférence
API Pipeline
Générer l'API
Optimisation
Discuter avec des modèles
Mise en production
Entraînement
Commencer
Personnalisation
Performance
Entraînement distribué
Matériel
Quantification
AperçuConcepts de quantificationChoisir une méthode OptimumContribuer une méthode de quantification
Tâches courantes
Quantifier lors du chargement
Charger des points de contrôle pré-quantifiés
Intégrations à l'écosystème
Ressources
API
You are viewing main version, which requires installation from source. If you'd like
regular pip install, checkout the latest stable version (v5.22.2).
SpQR
L'algorithme de quantification SpQR repose sur une structure de quantification 3 bits en groupes bi-niveaux, organisée en tuiles de 16x16, avec des valeurs aberrantes rares.

Pour quantiser un modèle avec SpQR, reportez-vous au dépôt Vahe1994/SpQR.
Charger un modèle quantifié SpQR avec from_pretrained().
from transformers import AutoTokenizer, AutoModelForCausalLM
import torch
quantized_model = AutoModelForCausalLM.from_pretrained(
"elvircrn/Llama-2-7b-SPQR-3Bit-16x16-red_pajama-hf",
dtype=torch.half,
device_map="auto"
)
tokenizer = AutoTokenizer.from_pretrained("elvircrn/Llama-2-7b-SPQR-3Bit-16x16-red_pajama-hf")