Views
No views yet
1apt update
2apt install git-lfs vim -y
3
4mkdir -p ~/miniconda3
5wget https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh -O ~/miniconda3/miniconda.sh
6bash ~/miniconda3/miniconda.sh -b -u -p ~/miniconda3
7~/miniconda3/bin/conda init bash
8source ~/.bashrc
9
10conda create -n hqq python=3.10 -y && conda activate hqq
11
12git lfs install
13git clone https://github.com/mobiusml/hqq.git
14cd hqq
15
16pip install torch
17pip install .
18
19pip install huggingface_hub[hf_transfer]
20export HF_HUB_ENABLE_HF_TRANSFER=1
21
22huggingface-cli loginquantize.py file by copy/pasting this into console:echo "
import torch
model_id = 'meta-llama/Meta-Llama-3-70B-Instruct'
save_dir = 'cat-llama-3-70b-hqq'
compute_dtype = torch.bfloat16
from hqq.core.quantize import *
quant_config = BaseQuantizeConfig(nbits=4, group_size=64, offload_meta=True)
zero_scale_group_size = 128
quant_config['scale_quant_params']['group_size'] = zero_scale_group_size
quant_config['zero_quant_params']['group_size'] = zero_scale_group_size
from hqq.engine.hf import HQQModelForCausalLM, AutoTokenizer
model = HQQModelForCausalLM.from_pretrained(model_id)
from hqq.models.hf.base import AutoHQQHFModel
AutoHQQHFModel.quantize_model(model, quant_config=quant_config,
compute_dtype=compute_dtype)
AutoHQQHFModel.save_quantized(model, save_dir)
model = AutoHQQHFModel.from_quantized(save_dir)
model.eval()
" > quantize.pypython quantize.py