Views
No views yet
1git clone https://github.com/RanchiZhao/AutoGPTQ.git && cd AutoGPTQ
2git checkout minicpmo
3
4# install AutoGPTQ
5pip install -vvv --no-build-isolation -e .AutoGPTQForCausalLM.from_quantized1import torch
2from transformers import AutoModel, AutoTokenizer
3from auto_gptq import AutoGPTQForCausalLM
4
5model = AutoGPTQForCausalLM.from_quantized(
6 'openbmb/MiniCPM-o-2_6-int4',
7 torch_dtype=torch.bfloat16,
8 device="cuda:0",
9 trust_remote_code=True,
10 disable_exllama=True,
11 disable_exllamav2=True
12)
13tokenizer = AutoTokenizer.from_pretrained(
14 'openbmb/MiniCPM-o-2_6-int4',
15 trust_remote_code=True
16)
17
18model.init_tts()
19