Views
No views yet
OpenOneRec/OneReason-8B-pretrain-competition.
It targets R0 SID-to-caption generation for the OneReason recommendation
competition.1e-4, cosine schedule, 3% warmup1.9813 versus 1.8500 for the RL10 parent:| Scope | Paired delta vs RL10 |
|---|---|
| Overall | +0.1313 |
| Short video | +0.2200 |
| Product | +0.2500 |
| Advertisement | +0.0100 |
| Livestream | +0.0450 |
[+0.0475,+0.2163], with win/tie/loss 161/127/112.-0.006465 and R3
restricted s_c probability -0.008117. Formal platform evaluation has not
been performed for this adapter.4c80e5ffd6ab9ae25e610df28e4dada30d64de771ac8145136d378d9c836e1ce1import torch
2from peft import PeftModel
3from transformers import AutoModelForCausalLM, AutoTokenizer
4
5base_id = "OpenOneRec/OneReason-8B-pretrain-competition"
6adapter_id = "yujackein/onereason-8b-lora-r0-sid-caption-residual-step65-drop24-35"
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)