Views
No views yet
Diamegs/PIT-4B-FT-202212
on synthetic + natural earnings-call QA. It is the SFT stage that precedes
the DPO model at
jdecim/PIT-4B-FT-202212-earnings-SFT-DPO.Diamegs/PIT-4B-FT-202212 (4B params, decoder-only, 2048-token context)jdecim/pit-earnings-call-qa
splits 202212/sft_train.jsonl / sft_val.jsonl (forward-synthetic,
forward-natural, inverse-natural, unanswerable QA, ~70K examples).<|user|> / <|assistant|> / <|end|> tags as the PIT base:<|user|>
Context:
{context}
Question: {question}
<|assistant|>
{answer}<|end|>1from peft import PeftModel
2from transformers import AutoModelForCausalLM, AutoTokenizer
3import torch
4
5base = "Diamegs/PIT-4B-FT-202212"
6adapter = "jdecim/PIT-4B-FT-202212-earnings-SFT"
7
8tok = AutoTokenizer.from_pretrained(base, trust_remote_code=True)
9model = AutoModelForCausalLM.from_pretrained(
10 base, torch_dtype=torch.bfloat16, trust_remote_code=True, device_map="auto",
11)
12model = PeftModel.from_pretrained(model, adapter)
13model.eval()
14
15prompt = (
16 "<|user|>\nContext:\nRevenue was $641 million, decreased by 10%.\n\n"
17 "Question: What was revenue?\n<|assistant|>\n"
18)
19out = model.generate(**tok(prompt, return_tensors="pt").to(model.device),
20 max_new_tokens=128, do_sample=False)
21print(tok.decode(out[0], skip_special_tokens=False).split("<|end|>")[0])acc_norm where the task defines it, acc otherwise — the
standard lm-eval-harness convention.| Task | Base PIT-4B-FT-202212 | + Earnings SFT (this model) |
|---|---|---|
| HellaSwag (acc_norm) | 0.645 | 0.659 |
| PIQA (acc_norm) | 0.774 | 0.758 |
| Winogrande (acc) | 0.634 | 0.618 |
| BoolQ (acc) | 0.609 | 0.634 |
| ARC-Easy (acc_norm) | 0.618 | 0.550 |
| ARC-Challenge (acc_norm) | 0.347 | 0.317 |
| OpenBookQA (acc_norm) | 0.396 | 0.366 |
| General avg | 0.575 | 0.557 |
| IFEval (prompt-level strict) | 0.146 | 0.165 |
| GSM8K (flex-extract) | 0.017 | 0.021 |
| MMLU (acc) | 0.265 | 0.239 |
jdecim/pit-earnings-call-qa @ 202212/benchmark_1k.jsonl)Pending. To be filled after runningrunai/run_benchmark_eval.sh(seesrc/evaluate_benchmark_metrics.py). Per-bucket metrics:forward_synthetic(number-recall, token-F1, exact-match),unanswerable(refusal-accuracy, hallucination-rate),forward_natural/inverse_natural(has-content, loop-rate, topic-F1).
Diamegs/PIT-4B-FT-202212.