Transformers documentation
SpQR
Comenzar
Clases base
Modelos
Preprocesadores
Inferencia
API de Pipeline
Generar API
Optimización
Chatear con modelos
Servicio
Entrenamiento
Hardware
Cuantización
ResumenConceptos de cuantizaciónElija un método OptimumContribuir con un método de cuantización
Tareas comunes
Cuantizar al cargar
Cargar puntos de control pre-cuantizados
Integraciones del ecosistema
Recursos
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
El algoritmo de cuantización SpQR implica una estructura de cuantización de 3 bits con grupos de dos niveles en mosaicos de 16x16 y valores atípicos dispersos.

Para cuantizar un modelo con SpQR, consulte el repositorio Vahe1994/SpQR.
Cargue un modelo cuantizado con SpQR mediante 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")