Views
No views yet
合并后的完整模型请前往 lemonbucket/yuwen-v3
| 项目 | 说明 |
|---|---|
| 模型 | Qwen/Qwen3.5-0.8B |
| 参数量 | ~0.8B |
| 架构 | Qwen3.5 (混合线性注意力 + 全注意力) |
| 参数 | 值 |
|---|---|
| 微调方式 | LoRA (PEFT) |
| 秩 (r) | 64 |
| Alpha | 128 |
| Dropout | 0.1 |
| 目标模块 | q_proj, k_proj, v_proj, o_proj, gate_proj, up_proj, down_proj |
| 任务类型 | Causal LM |
| 训练轮数 | 8 Epochs (376 steps) |
| 批次大小 | 1 × 8 (梯度累积) |
| 学习率 | 1e-4 |
| 预热比例 | 5% |
| 最大序列长度 | 1024 |
| 精度 | F32 |
| 设备 | Apple Silicon MPS |
注: 从 Step 90 后验证损失开始上升 (0.28→0.30→0.32),说明第 2 个 Epoch 后开始过拟合。
1from transformers import AutoModelForCausalLM, AutoTokenizer
2from peft import PeftModel
3
4# 加载基座模型
5base_model = AutoModelForCausalLM.from_pretrained(
6 "Qwen/Qwen3.5-0.8B",
7 torch_dtype="auto",
8 trust_remote_code=True,
9)
10tokenizer = AutoTokenizer.from_pretrained("Qwen/Qwen3.5-0.8B", trust_remote_code=True)
11
12# 加载 LoRA 适配器
13model = PeftModel.from_pretrained(base_model, "lemonbucket/yuwen-v3-lora")
14
15# 推理
16messages = [
17 {"role": "system", "content": "你是yuwen-v3,由柠檬桶基于Qwen3.5-0.8B微调的语文模型,擅长语文阅读理解答题。回答要分点作答、术语规范、踩分精准。"},
18 {"role": "user", "content": "请分析《荷塘月色》中通感手法的运用。"},
19]
20text = tokenizer.apply_chat_template(messages, tokenize=False, add_generation_prompt=True)
21inputs = tokenizer(text, return_tensors="pt")
22outputs = model.generate(**inputs, max_new_tokens=1024, temperature=0.3, top_k=20, top_p=0.85)
23print(tokenizer.decode(outputs[0], skip_special_tokens=True))