SpiridonSunRotator's picture
Added evaluation metrics
3da669d verified
---
library_name: transformers
tags:
- aqlm
- llama
- facebook
- meta
- llama-3
- conversational
- text-generation-inference
base_model: meta-llama/Llama-3.2-1B
---
Official [AQLM](https://arxiv.org/abs/2401.06118) quantization of [meta-llama/Llama-3.2-1B
](https://huggingface.co/meta-llama/Llama-3.2-1B) finetuned with [PV-Tuning](https://arxiv.org/abs/2405.14852).
For this quantization, we used 2 codebooks of 8 bits and groupsize of 8.
Results:
| Model | Quantization | MMLU (5-shot) | ArcC | ArcE | Hellaswag | PiQA | Winogrande | Model size, Gb |
|-------|--------------|---------------|--------|--------|-----------|--------|------------|----------------|
| meta-llama/Llama-3.2-1B | fp16 | 0.3195 | 0.3123 | 0.6553 | 0.4772 | 0.7448 | 0.6054 | 2.5 |
| | 2x8g8 | 0.2465 | 0.2713 | 0.5896 | 0.4034 | 0.7067 | 0.5564 | 0.8 |