README.md

---
language:
- zh
pipeline_tag: text-generation
tags:
- pytorch
- causal-lm
- custom-code
---

# MiniSnail DPO

MiniSnail 是一个从头实现和训练的轻量级自回归语言模型。

## 模型结构

- Vocabulary size: 6,400
- Context length: 512
- Hidden size: 768
- Layers: 12
- Attention heads: 12
- FFN size: 2,048
- Training stages: Pre-training → SFT → DPO

## 使用限制

该模型采用自定义 PyTorch 架构，目前不能直接通过
`transformers.AutoModel.from_pretrained()` 加载。

模型规模较小，可能出现事实错误、重复输出、不完整回答和不安全内容。
不应直接用于医疗、法律、金融等高风险场景。

## 对话格式

```text
<|im_start|>user
问题<|im_end|>
<|im_start|>assistant