How to use from the
Use from the
Transformers library
# Use a pipeline as a high-level helper
from transformers import pipeline

pipe = pipeline("text-generation", model="summerAIO/Looped-Qwen-pre", trust_remote_code=True)
messages = [
    {"role": "user", "content": "Who are you?"},
]
pipe(messages)
# pip install -U transformers accelerate
# Load model directly
from transformers import AutoModelForCausalLM
model = AutoModelForCausalLM.from_pretrained("summerAIO/Looped-Qwen-pre", trust_remote_code=True, device_map="auto")
Quick Links

summerMC/Looped-Qwen-pre

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

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

Model tree for summerAIO/Looped-Qwen-pre

Finetuned
Qwen/Qwen3.5-2B
Finetuned
(654)
this model
Finetunes
2 models