Looped-Qwen-gsm8k

summerMC/Looped-Qwen-pre (Looped Gated DeltaNet, ็ด„ 1B) ใ‚’่ฟฝๅŠ ๅญฆ็ฟ’ใ—ใ€ LoRA ใ‚’ๆœฌไฝ“ใซใƒžใƒผใ‚ธใ—ใŸ bf16 ็‰ˆใงใ™ใ€‚่ชญใฟ่พผใฟใซใฏ trust_remote_code=True ใŒๅฟ…่ฆใงใ™ใ€‚

่ฉ•ไพก (GSM8K test ๅ…ˆ้ ญ 200 ๅ•, ็ญ”ใˆ้ƒจๅˆ†ใฎ teacher-forcing loss)

ใƒขใƒ‡ใƒซ loss ppl
summerMC/Looped-Qwen-pre (ใƒ™ใƒผใ‚น) 2.3437 10.42
ใ“ใฎใƒขใƒ‡ใƒซ 1.1382 3.12

ๆญฃ่งฃ็އ (็”Ÿๆˆใ—ใฆ็ญ”ใˆใ‚’็…งๅˆ) ใงใฏใชใใ€ๆญฃ่งฃใฎ็ญ”ใˆ้ƒจๅˆ†ใซๅฏพใ™ใ‚‹ loss ใงใ™ใ€‚

ไฝฟใ„ๆ–น

import torch
from transformers import AutoModelForCausalLM, AutoTokenizer
repo = "summermc/Looped-Qwen-gsm8k"
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": "Janet has 16 eggs. She eats 3 and bakes with 4. How many are left?"}],
                              add_generation_prompt=True, return_tensors="pt", return_dict=True,
                              enable_thinking=False)["input_ids"].cuda()
out = model.generate(ids, max_new_tokens=256, 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 ๆŽจๅฅจใ€‚greedy ็”ŸๆˆใงใฏๅŒใ˜่กŒใ‚’็นฐใ‚Š่ฟ”ใ—ใ‚„ใ™ใ„ใฎใงใ€ repetition_penalty ใ‚’ไป˜ใ‘ใŸ sampling ใ‚’ๆŽจๅฅจใ—ใพใ™ใ€‚

ๆณจๆ„

  • ๅฐๅž‹ใƒขใƒ‡ใƒซใฎใŸใ‚่จˆ็ฎ—ใƒŸใ‚นใ‚„็นฐใ‚Š่ฟ”ใ—ใŒ่ตทใใพใ™ใ€‚
  • ใƒฉใ‚คใ‚ปใƒณใ‚นใฏๅ…ƒใƒขใƒ‡ใƒซ (Qwen/Qwen3.5-2B, Apache-2.0) ใซๅพ“ใ„ใพใ™ใ€‚
  • ๅ…ƒใƒขใƒ‡ใƒซ: Qwen Team / ๅ…จ GDN ๅŒ–ใƒปLooped ๅŒ–: summerMC, j-llm
Downloads last month
350
Safetensors
Model size
1.0B params
Tensor type
BF16
ยท
Inference Providers NEW
This model isn't deployed by any Inference Provider. ๐Ÿ™‹ Ask for provider support

Model tree for summerAIO/Looped-Qwen-gsm8k

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