Views
No views yet
pip install transformers==4.56.0 torch==2.9.1 auto_round==0.9.41import torch
2from transformers import AutoModelForCausalLM, AutoTokenizer
3from auto_round import AutoRound
4
5model_name = "MiniMaxAI/MiniMax-M2.5"
6
7model = AutoModelForCausalLM.from_pretrained(model_name, device_map="cpu", trust_remote_code=True, dtype="auto")
8tokenizer = AutoTokenizer.from_pretrained(model_name)
9layer_config = {}
10for n, m in model.named_modules():
11 if n == "lm_head" or isinstance(m,torch.nn.Embedding):
12 layer_config[n] = {"bits": 8}
13 elif isinstance(m, torch.nn.Linear) and (not "expert" in n or "shared_experts" in n) and n != "lm_head":
14 layer_config[n] = {"bits": 4}
15
16autoround = AutoRound(model, tokenizer, iters=0, layer_config=layer_config, nsamples=512, disable_opt_rtn=False)
17
18autoround.quantize_and_save("/models/tmp_autoround", format="gguf:q2_k_s")