Views
No views yet
| Parameter | Value |
|---|---|
| Base model | Qwen/Qwen2.5-1.5B |
| Method | Supervised Fine-Tuning (SFT) via LoRA |
| LoRA rank | 4 |
| LoRA alpha | 32 |
| LoRA target modules | q_proj, k_proj, v_proj, o_proj, gate_proj, up_proj, down_proj |
| Learning rate | 1e-5 |
| Batch size | 4 |
| Epochs | 1 |
| Training data | GSM8K (math word problems) + CommonsenseQA (commonsense QA) |
| Framework | TRL SFTTrainer + PEFT |
| Metric | Base | SFT-1 (this model) |
|---|---|---|
| LLM Judge Accuracy | 94.4% | 92.6% |
| Exact Match | 59.3% | 53.7% |
| BERTScore F1 | 0.7645 | 0.7772 |
| BLEU | 0.0307 | 0.0373 |
| Answer Stability Rate | 55.6% | 64.8% |
| Distractor Degradation | 16.7% | 11.1% |
| Reasoning Judge Score (1–5) | 4.69 | 4.85 |
1from transformers import AutoModelForCausalLM, AutoTokenizer
2import torch
3
4model = AutoModelForCausalLM.from_pretrained(
5 "syedtaha22/Qwen2.5-1.5B-GSM8K-CommonsenseQA-SFT",
6 torch_dtype=torch.bfloat16,
7 device_map="auto",
8)
9tokenizer = AutoTokenizer.from_pretrained(
10 "syedtaha22/Qwen2.5-1.5B-GSM8K-CommonsenseQA-SFT"
11)
12
13prompt = "A baker makes 48 cookies. She sells 1/3 of them in the morning and half of the remainder in the afternoon. How many cookies does she have left?"
14inputs = tokenizer(prompt, return_tensors="pt").to(model.device)
15
16with torch.no_grad():
17 output = model.generate(**inputs, max_new_tokens=256, temperature=0.1, do_sample=True)
18
19print(tokenizer.decode(output[0], skip_special_tokens=True))1@misc{qwen2025qwen25technicalreport,
2 title={Qwen2.5 Technical Report},
3 author={Qwen et al.},
4 year={2025},
5 eprint={2412.15115},
6 archivePrefix={arXiv},
7}