AI-MO/NuminaMath-CoT
Viewer • Updated • 860k • 30.3k • 593
How to use ermiaazarkhalili/LFM2.5-1.2B-Instruct-GRPO-NuminaMath-10K with PEFT:
from peft import PeftModel
from transformers import AutoModelForCausalLM
base_model = AutoModelForCausalLM.from_pretrained("LiquidAI/LFM2.5-1.2B-Instruct")
model = PeftModel.from_pretrained(base_model, "ermiaazarkhalili/LFM2.5-1.2B-Instruct-GRPO-NuminaMath-10K")GRPO (Group Relative Policy Optimization) LoRA fine-tune of
LiquidAI/LFM2.5-1.2B-Instruct on the
AI-MO/NuminaMath-CoT dataset
(10K training samples).
| Field | Value |
|---|---|
| Base model | LiquidAI/LFM2.5-1.2B-Instruct |
| Method | GRPO (RL) with LoRA/PEFT adapters |
| Dataset | AI-MO/NuminaMath-CoT |
| Training samples | 10K |
| Adapter | LoRA — load on top of the base model with PEFT |
from peft import PeftModel
from transformers import AutoModelForCausalLM, AutoTokenizer
base = AutoModelForCausalLM.from_pretrained("LiquidAI/LFM2.5-1.2B-Instruct", torch_dtype="auto", device_map="auto")
model = PeftModel.from_pretrained(base, "ermiaazarkhalili/LFM2.5-1.2B-Instruct-GRPO-NuminaMath-10K")
tok = AutoTokenizer.from_pretrained("ermiaazarkhalili/LFM2.5-1.2B-Instruct-GRPO-NuminaMath-10K")
msgs = [{"role": "user", "content": "Solve: what is 17 * 24?"}]
prompt = tok.apply_chat_template(msgs, tokenize=False, add_generation_prompt=True)
out = model.generate(**tok(prompt, return_tensors="pt").to(model.device), max_new_tokens=512)
print(tok.decode(out[0], skip_special_tokens=True))
Research / non-commercial experimentation on mathematical reasoning. As a LoRA
adapter it inherits every limitation of the base model; verify outputs before use.
A GGUF build (adapter merged onto the base) is available at ermiaazarkhalili/LFM2.5-1.2B-Instruct-GRPO-NuminaMath-10K-GGUF.
Base model
LiquidAI/LFM2.5-1.2B-Base