Views
No views yet
OpenOneRec/OneReason-8B-pretrain-competition,
trained for the OneReason recommendation competition.| Category | Source rows | Source rendered tokens |
|---|---|---|
| R0 perception | 32,000 | 5,801,468 |
| R2 evolution | 24,636 | 119,130,968 |
| R3 recommendation | 24,134 | 34,445,219 |
| Clean C-Eval world data | 1,601 | 195,253 |
| Total | 82,371 | 159,572,908 |
R2_evolution rows longer than the cutoff were
removed because head truncation would eliminate all assistant supervision.
The resulting training set has 82,356 rows and 159,413,369 rendered tokens.
The exact four-rank packing plan consumes 99.0516% of these tokens in one safe
pass. All assistant loss weights were normalized to 1.0.r=32, alpha=32, dropout 0.05)q_proj, k_proj, v_proj, o_proj, gate_proj, up_proj, down_proj2e-4, cosine schedule, 3% warmupec65db55748eeb8181591b53ba1d2eb7f3dfea702e5e32552f93e6198f45a776yujackein/onereason-8b-lora-item32k-user75-rec50-worldclean1601-all1-lr2e4-r32a32-step323,
the one-packing-pass checkpoint from the 32,768-context r32/a32 LR2e-4 run.1import torch
2from peft import PeftModel
3from transformers import AutoModelForCausalLM, AutoTokenizer
4
5base_id = "OpenOneRec/OneReason-8B-pretrain-competition"
6adapter_id = "yujackein/onereason-8b-lora-item32k-user75-rec50-worldclean1601-all1-lr2e4-r32a32-seq10k-gbs4-1epoch"
7
8tokenizer = AutoTokenizer.from_pretrained(adapter_id, trust_remote_code=True)
9base_model = AutoModelForCausalLM.from_pretrained(
10 base_id,
11 torch_dtype=torch.bfloat16,
12 device_map="auto",
13 trust_remote_code=True,
14)
15model = PeftModel.from_pretrained(base_model, adapter_id)