Qwen3.5-2B-Longbook2 (accuracy-recovered)

ๅค‰ๆ›ๅ…ƒใƒขใƒ‡ใƒซ ใฏ Qwen3.5-2B ใฎๅ…จ full-attention ๅฑคใ‚’ Gated DeltaNet (GDN) ใซ็ฝฎใๆ›ใˆใŸ all-GDN ใƒขใƒ‡ใƒซใงใ™ใ€‚็ฝฎๆ›ใง่ฝใกใŸ็ฒพๅบฆใ‚’ใ€ ๅ…ƒใƒขใƒ‡ใƒซ Qwen/Qwen3.5-2B ใ‚’ๆ•™ๅธซใจใ—ใŸ่’ธ็•™ใงๅ›žๅพฉใ—ใŸใ‚‚ใฎใŒใ“ใฎใƒชใƒใ‚ธใƒˆใƒชใงใ™ใ€‚ ่ฟฝๅŠ ใฎ็Ÿฅ่ญ˜ใƒ‡ใƒผใ‚ฟใงใฎ SFT ใฏ่กŒใฃใฆใ„ใพใ›ใ‚“(ๅพฉๅ…ƒใฎใฟ)ใ€‚

  • ๅ…จๅฑคใŒ็ทšๅฝขใ‚ขใƒ†ใƒณใ‚ทใƒงใƒณ โ†’ KV ใ‚ญใƒฃใƒƒใ‚ทใƒฅใŒๅข—ใˆใšใ€้•ทๆ–‡ใงใ‚‚ใƒกใƒขใƒชใƒป้€ŸๅบฆใŒไธ€ๅฎš
  • ๆจ™ๆบ–ใฎ transformers ใงใใฎใพใพ่ชญใฟ่พผใ‚ใพใ™(trust_remote_code ไธ่ฆ)
  • flash-linear-attention ใ‚’ๅ…ฅใ‚Œใ‚‹ใจ้ซ˜้€Ÿใ‚ซใƒผใƒใƒซใŒไฝฟใ‚ใ‚Œใพใ™

ๅพฉๅ…ƒๆ–นๆณ•

  1. Stage 1 (ๅฑคใ”ใจใฎ่’ธ็•™): ็ฝฎๆ›ใ—ใŸๅ„ GDN ๅฑคใฎๅ‡บๅŠ›ใ‚’ๅ…ƒใฎ full-attention ๅฑคใฎๅ‡บๅŠ›ใซๅˆใ‚ใ›ใ‚‹ (็›ธๅฏพ MSE + ใ‚ณใ‚ตใ‚คใƒณ)ใ€‚ๅฑค้–“ใฎๅ…ฅๅŠ›ใฏ detach ใ—ใ€ๅ…จๅฑคใ‚’ไธฆๅˆ—ใซๅญฆ็ฟ’ใ€‚
  2. Stage 2 (end-to-end): ๆ•™ๅธซ logits ใธใฎ KL + ้š ใ‚Œ็Šถๆ…‹ๆๅคฑใงๅ…จไฝ“ใ‚’ๅพฎ่ชฟๆ•ดใ€‚
  3. ๅญฆ็ฟ’ใ—ใŸ้‡ใฟใ‚’ๅ…ƒใฎ safetensors ใ‚ทใƒฃใƒผใƒ‰ใฎ่ฉฒๅฝ“ใƒ†ใƒณใ‚ฝใƒซใ ใ‘ๅทฎใ—ๆ›ฟใˆใฆๅ‡บๅŠ›ใ€‚

fp32 ใƒžใ‚นใ‚ฟใƒผ้‡ใฟ + bf16 autocastใ€fused AdamWใ€‚

็ตๆžœ

(่ฉ•ไพกๅ€คใฏๆœช่จ˜ๅ…ฅ)

(ๅŒไธ€ใฎ่ฉ•ไพกใƒ†ใ‚ญใ‚นใƒˆใƒปๅŒไธ€ๆกไปถใงๆธฌๅฎšใ€‚ๆ•™ๅธซใฏ Qwen/Qwen3.5-2B)

ไฝฟใ„ๆ–น

import torch
from transformers import AutoModelForCausalLM, AutoTokenizer
repo = "j-llm/Qwen3.5-2B-Longbook2"
tok = AutoTokenizer.from_pretrained(repo)
model = AutoModelForCausalLM.from_pretrained(repo, dtype=torch.bfloat16, device_map="cuda").eval()
ids = tok.apply_chat_template([{"role": "user", "content": "ๆ—ฅๆœฌใฎ้ฆ–้ƒฝใฏใฉใ“๏ผŸ"}],
        add_generation_prompt=True, return_tensors="pt", return_dict=True, enable_thinking=False).to(model.device)
out = model.generate(**ids, max_new_tokens=128, use_cache=False)   # ไธ‹ใฎๆณจๆ„ๅ‚็…ง
print(tok.decode(out[0, ids["input_ids"].shape[1]:], skip_special_tokens=True))

ๆณจๆ„

  • ๅ…จๅฑคใŒ็ทšๅฝขใ‚ขใƒ†ใƒณใ‚ทใƒงใƒณใฎใŸใ‚ใ€็พ่กŒ transformers ใฎๆจ™ๆบ– Cache ใฏ get_seq_length ใง ValueError ใ‚’ๅ‡บใ™ใ“ใจใŒใ‚ใ‚Šใพใ™ใ€‚ใใฎๅ ดๅˆใฏ use_cache=False(้…ใ„ใŒ็ขบๅฎŸ)ใซใ™ใ‚‹ใ‹ใ€ get_seq_length ใ‚’ไพ‹ๅค–ๆ™‚ใซ่‡ชๅ‰ใ‚ซใ‚ฆใƒณใ‚ฟใธใƒ•ใ‚ฉใƒผใƒซใƒใƒƒใ‚ฏใ•ใ›ใ‚‹ใƒ‘ใƒƒใƒใ‚’ๅฝ“ใฆใฆใใ ใ•ใ„ใ€‚
  • ่’ธ็•™ใฏๆ•™ๅธซใฎๆŒ™ๅ‹•ใธใฎ่ฟ‘ไผผใงใ‚ใ‚Šใ€ๅ…ƒใƒขใƒ‡ใƒซใจๅฎŒๅ…จใซใฏไธ€่‡ดใ—ใพใ›ใ‚“ใ€‚็‰นใซๆŽจ่ซ–ใƒปๆ•ฐๅญฆใƒปใ‚ณใƒผใƒ‰ใฏๅ…ƒใƒขใƒ‡ใƒซใ‚ˆใ‚ŠๅŠฃใ‚‹ๅฏ่ƒฝๆ€งใŒใ‚ใ‚Šใพใ™ใ€‚
  • ใƒฉใ‚คใ‚ปใƒณใ‚นใฏๅ…ƒใƒขใƒ‡ใƒซ(Qwen/Qwen3.5-2B)ใซๅพ“ใ„ใพใ™ใ€‚

่ฌ่พž

ๅค‰ๆ›ใƒขใƒ‡ใƒซ: summerMC / ๅ…ƒใƒขใƒ‡ใƒซ: Qwen Team

Downloads last month
306
Safetensors
Model size
2B params
Tensor type
F32
ยท
BF16
ยท
Inference Providers NEW
This model isn't deployed by any Inference Provider. ๐Ÿ™‹ Ask for provider support

Model tree for j-llm/Qwen3.5-2B-Longbook2

Finetuned
Qwen/Qwen3.5-2B
Finetuned
(638)
this model
Quantizations
1 model

Collection including j-llm/Qwen3.5-2B-Longbook2