manueldeprada HF Staff commited on
Commit
5709ad2
·
verified ·
1 Parent(s): 56164a1

Upload README.md with huggingface_hub

Browse files
Files changed (1) hide show
  1. README.md +45 -0
README.md ADDED
@@ -0,0 +1,45 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: other
3
+ language: en
4
+ tags:
5
+ - littlelearner
6
+ - qwen3
7
+ - k5
8
+ ---
9
+
10
+ # littlelearner-5b-bounded-sft-chatty
11
+
12
+ The chat-behavior variant of the LittleLearner 5B bounded model. Base =
13
+ `littlelearner-5b-bounded-base` (5.04B Qwen3-dense, 88B tokens of K-5 LittleCurriculum with the 5%
14
+ cooloff SFT blend), followed by one epoch of behavior SFT with fp32 master parameters (lr 1e-5) on a
15
+ mix of K-5 math CoT (30k), smoltalk K-5 general chat (15k), GSM8K K-5 (6.9k), format-control pairs
16
+ with varied system prompts (16.3k), and LittleLearner identity data (16.7k).
17
+
18
+ Compared to `littlelearner-5b-bounded-sft`, this model adds: casual chat (greetings answered
19
+ conversationally instead of with a math word problem), a self-identity ("I am LittleLearner, a small
20
+ language model. I was trained on kindergarten-through-grade-5 school material."), and instruction
21
+ steerability: "answer with only the final number" shortens replies from ~270 to ~11 characters
22
+ (obedience 0.90 on a held-out instruction probe, 0.85 via a held-out system prompt).
23
+
24
+ Filtered chat MathCAMPS (K-5 grades 2-5 mean): pass@64 78.7, first-sample pass@1 31.4 (the non-chatty
25
+ sft model: 80.1 / 45.7 under a slightly different eval condition). Answer-only obedience strips
26
+ chain-of-thought, which lowers accuracy on hard problems; ask for steps when accuracy matters. This
27
+ checkpoint replaces an earlier lr 3e-6 variant whose identity installation was partial.
28
+
29
+ ## Usage
30
+
31
+ ```python
32
+ from transformers import AutoModelForCausalLM, AutoTokenizer
33
+
34
+ repo = "littlelearner/littlelearner-5b-bounded-sft-chatty"
35
+ tok = AutoTokenizer.from_pretrained(repo)
36
+ model = AutoModelForCausalLM.from_pretrained(repo, dtype="bfloat16", device_map="auto")
37
+ msgs = [{"role": "user", "content": "If Sarah has 12 apples and gives 5 to Tom, how many does she have left?"}]
38
+ ids = tok.apply_chat_template(msgs, add_generation_prompt=True, return_tensors="pt").to(model.device)
39
+ out = model.generate(ids)
40
+ print(tok.decode(out[0, ids.shape[1]:], skip_special_tokens=True))
41
+ ```
42
+
43
+ Part of the LittleLearner study (pedagogically-controlled knowledge exposure). The bounded corpus is
44
+ FineWeb-Edu filtered to U.S. K-5 material; see the org's other repos for the unbounded controls and
45
+ GRPO variants.