File size: 372 Bytes
8f21213 7ce67d3 8f21213 7ce67d3 8f21213 7ce67d3 8f21213 7ce67d3 8f21213 7ce67d3 8f21213 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 | # payelb/PKUSafeRLHF_TinyLlama-1.1B_aligned_with_MARS_RM
Base model: TinyLlama/TinyLlama-1.1B-Chat-v1.0
Alignment dataset: PKU-Alignment/PKU-SafeRLHF
Reward model: payelb/PKUSafeRLHF_roberta-base_1k_fixed_MARS
Method: PPO alignment with LoRA adapters.
Notes:
- Reward normalization and clipping enabled
- KL control enabled
- pad_token_id/eos_token_id explicitly set
|