How to use from the
Use from the
Transformers library
# Use a pipeline as a high-level helper
from transformers import pipeline

pipe = pipeline("text-generation", model="ISTA-DASLab/Meta-Llama-3-8B-AQLM-2Bit-1x16")
messages = [
    {"role": "user", "content": "Who are you?"},
]
pipe(messages)
# Load model directly
from transformers import AutoTokenizer, AutoModelForCausalLM

tokenizer = AutoTokenizer.from_pretrained("ISTA-DASLab/Meta-Llama-3-8B-AQLM-2Bit-1x16")
model = AutoModelForCausalLM.from_pretrained("ISTA-DASLab/Meta-Llama-3-8B-AQLM-2Bit-1x16", device_map="auto")
Quick Links

Official AQLM quantization of meta-llama/Meta-Llama-3-8B .

For this quantization, we used 1 codebook of 16 bits.

Results:

Model Quantization MMLU (5-shot) ArcC ArcE Hellaswag PiQA Winogrande Model size, Gb
meta-llama/Meta-Llama-3-8B None 0.6525 0.5043 0.8009 0.6019 0.7971 0.7261 16.1
1x16 0.5647 0.4130 0.7428 0.5545 0.7780 0.7198 4.1

UPD 02.05.2024

The version of model with improved fine-tuning procedure.

Downloads last month
26
Safetensors
Model size
2B params
Tensor type
F16
·
I16
·
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support

Collection including ISTA-DASLab/Meta-Llama-3-8B-AQLM-2Bit-1x16

Paper for ISTA-DASLab/Meta-Llama-3-8B-AQLM-2Bit-1x16