MiniMind3-Ascend-Dense
这是一个基于 MiniMind3-Ascend 训练链路导出的 Dense 对话模型,默认兼容 Transformers 推理方式,适合作为轻量级中文对话模型使用。
模型信息
- 模型名:
fzkun/minimind3-ascend-dense - 架构:Dense
- 导出兼容:
Qwen3ForCausalLM - 参数规模:约 64M
- 主要配置:
hidden_size = 768num_hidden_layers = 8
文件说明
仓库中包含:
config.jsongeneration_config.jsonmodel.safetensorstokenizer.jsontokenizer_config.jsonspecial_tokens_map.jsonchat_template.jinja
使用方式
Transformers
from transformers import AutoTokenizer, AutoModelForCausalLM
model_id = "fzkun/minimind3-ascend-dense"
tokenizer = AutoTokenizer.from_pretrained(model_id, trust_remote_code=True)
model = AutoModelForCausalLM.from_pretrained(model_id, trust_remote_code=True)
messages = [{"role": "user", "content": "你好"}]
text = tokenizer.apply_chat_template(messages, tokenize=False, add_generation_prompt=True)
inputs = tokenizer(text, return_tensors="pt")
out = model.generate(**inputs, max_new_tokens=256)
print(tokenizer.decode(out[0][inputs["input_ids"].shape[1]:], skip_special_tokens=True))
Benchmark 结果
评测环境:
- Ascend 910B
- 单卡
npu:0 batch_size = 16
| ceval | cmmlu | arc | piqa | openbookqa | hellaswag | siqa |
|---|---|---|---|---|---|---|
| 22.66 | 25.04 | 28.66 | 51.85 | 25.60 | 28.73 | 32.60 |
说明:
ceval / cmmlu / arc / piqa / openbookqa / hellaswag使用acc_normsocial_iqa使用acc
补充说明
- Dense 版本更轻量,适合资源受限场景
- 对应 SwanLab 实验记录:https://swanlab.cn/@fzkun/MiniMind3/overview
- Downloads last month
- 153