justheuristic's picture
Update README.md
88653f0 verified
metadata
library_name: transformers
tags:
  - llama
  - facebook
  - meta
  - llama-2
  - conversational
  - text-generation-inference

An official quantization of meta-llama/Llama-2-7b using PV-Tuning on top of AQLM. For this quantization, we used 1 codebook of 16 bits for groups of 16 weights, totalling about 1.58 bits per weight.

The 1x16g16 models require aqlm inference library v1.1.6 or newer:

pip install aqlm[gpu,cpu]>=1.1.6

Model AQLM scheme WikiText 2 PPL Model size, Gb Hub link
Llama-2-7b 1x16 5.68 2.4 Link
Llama-2-7b 2x8 5.90 2.2 Link
Llama-2-7b (this) 1x16g16 9.21 1.7 Link
Llama-2-13b 1x16 5.05 4.1 Link
Llama-2-70b 1x16 3.78 18.8 Link

To learn more about the inference, as well as the information on how to quantize models yourself, please refer to the official GitHub repo. The original code for PV-Tuning can be found in the AQLM@pv-tuning branch.