Dedicated Endpoints

Run this model inference on single tenant GPU with unmatched speed and reliability at scale.

Learn more
Container

Run this model inference with full control and performance in your environment.

Learn more

Get help setting up a custom Dedicated Endpoints.

Talk with our engineer to get a quote for reserved GPU instances with discounts.

README

Results

True reproducible exact-match evaluation with deterministic greedy decoding and required #### <integer> answer extraction:

  • 20% GSM8K / 8% SVAMP
  • Evaluation subsets: GSM8K test first 100 examples, SVAMP first 50 examples
  • Decoding: greedy, max_new_tokens=256

Training-time metrics from the original run used non-reseeded random subsets and should not be used as final accuracy claims.

Intended Use

Use when GSM8K in-distribution accuracy matters most. This is a research checkpoint, not a production math solver.

Loading

python

from transformers import AutoTokenizer, AutoModelForCausalLM
from peft import PeftModel
import torch
base_id = "Qwen/Qwen2.5-1.5B-Instruct"
adapter_id = "ahmed-3m/qwen25-1.5b-gsm8k-reinforce-step425"
tok = AutoTokenizer.from_pretrained(base_id)
base = AutoModelForCausalLM.from_pretrained(
base_id,
torch_dtype=torch.float32,
attn_implementation="sdpa",
device_map="auto",
)
model = PeftModel.from_pretrained(base, adapter_id)
model.eval()

Training Context

  • Base model: Qwen2.5-1.5B-Instruct
  • Adapter: LoRA r=16 over projection modules
  • Task: GSM8K math reasoning
  • Hardware used: Tesla P40; fp16 + SDPA constraints

Model provider

ahmed-3m

Model tree

Base

Qwen/Qwen2.5-1.5B-Instruct

Adapter

this model

Modalities

Input

Text

Output

Text

Pricing

Dedicated Endpoints

View details

Supported Functionality

Model APIs

Dedicated Endpoints

Container

More information

Explore FriendliAI today