ISTA-DASLab
/

gemma-2b-AQLM-2Bit-1x16-hf

Text Generation

text-generation-inference

Inference Endpoints

Model card Files Files and versions Community

SpiridonSunRotator commited on Mar 10

Commit

509ed2d

•

1 Parent(s): cdeead5

Added fp16 baseline

Files changed (1) hide show

README.md +4 -3

README.md CHANGED Viewed

@@ -3,9 +3,10 @@ Official [AQLM](https://arxiv.org/abs/2401.06118) quantization of [google/gemma-
 For this quantization, we used 1 codebook of 16 bits.
-Results:
-| Model      | AQLM scheme | WinoGrande | PiQA | HellaSwag | ArcE | ArcC | Model size, Gb |
 |------|------|------|-------|-------|-------|------|------|
-| gemma-2b |1x16| 0.6275	| 0.7318	| 0.4582	| 0.6923	| 0.3259| 1.7 |
 To learn more about the inference, as well as the information on how to quantize models yourself, please refer to the [official GitHub repo](https://github.com/Vahe1994/AQLM).

 For this quantization, we used 1 codebook of 16 bits.
+Results (0-shot `acc`):
+| Model      | Quantization | WinoGrande | PiQA | HellaSwag | ArcE | ArcC | Model size, Gb |
 |------|------|------|-------|-------|-------|------|------|
+|gemma-2b| None | 0.6472 | 0.7715 | 0.5279 | 0.7403 | 0.4053| 5.0 |
+|  |1x16| 0.6275	| 0.7318	| 0.4582	| 0.6923	| 0.3259| 1.7 |
 To learn more about the inference, as well as the information on how to quantize models yourself, please refer to the [official GitHub repo](https://github.com/Vahe1994/AQLM).