summerMC/Looped-Qwen-pre2
j-llm/Qwen3.5-2B-SpeedX2 (全層 Gated DeltaNet) の 24 層を
空間マージして重み共有し、2 重ループにした Looped GDN (反復ステップ毎 LoRA 付き)。教師 Qwen/Qwen3.5-2B から蒸留。
for t in range(6): # 外側ループ
h = RoPE注入(h) # 位置の再注入 + ループ段埋め込み
for k in range(loops[t]): # 内側ループ (ゲート付き) loops = [4, 4, 4, 4, 4, 4]
h = h + g_k(h) * (GDN_t(h) - h) # GDN_t = 元の 4 層を平均マージした共有ブロック
使い方 (transformers)
import torch
from transformers import AutoModelForCausalLM, AutoTokenizer
repo = "summerMC/Looped-Qwen-pre"
tok = AutoTokenizer.from_pretrained(repo)
model = AutoModelForCausalLM.from_pretrained(repo, trust_remote_code=True, dtype=torch.bfloat16).cuda().eval()
ids = tok.apply_chat_template([{"role": "user", "content": "日本の首都はどこ？"}], add_generation_prompt=True,
return_tensors="pt", return_dict=True, enable_thinking=False)["input_ids"].cuda()
out = model.generate(ids, max_new_tokens=128, do_sample=True, temperature=0.7, top_p=0.9, repetition_penalty=1.1)
print(tok.decode(out[0, ids.shape[1]:], skip_special_tokens=True))
pip install -U transformers flash-linear-attention 推奨 (GDN 高速カーネル)。
ファイル
| ファイル | 内容 |
|---|---|
modeling_looped_gdn.py / configuration_looped_gdn.py | transformers 用カスタムコード (trust_remote_code) |
looped_gdn.py | モデル本体・空間マージ変換 (python looped_gdn.py --src j-llm/Qwen3.5-2B-SpeedX2 --groups 6) |
scripts/colab_looped_distill.py | 蒸留スクリプト (Colab に貼るだけ) |
scripts/infer_looped_gdn.py | 推論 / 対話スクリプト |
scripts/colab_ast_transplant.py | 高度な空間移植 (活性考慮 RegMean + DP 可変長分割 + 活性考慮 LoRA) |
scripts/test_looped_gdn.py | 単体テスト |
looped_config.json | 旧形式の設定 (LoopedGDNForCausalLM.load 用) |
学習
- Stage 1: グループ単位 hidden 蒸留 (相対 MSE + コサイン) / Stage 2: 教師 logits KL + hidden 損失
- データ: C4 (日本語・英語)
注意
蒸留による近似で教師より性能は低下します。ライセンスは元モデル (Qwen/Qwen3.5-2B) に従います。 元モデル: Qwen Team / 全 GDN 化: summerMC, j-llm