payelb's picture
Upload PPO-aligned TinyLlama-1.1B model using baseline DeBERTa reward model on UltraFeedback_openbmb
4a4e4b9 verified
|
raw
history blame
481 Bytes

payelb/UltraFeedback_openbmb_TinyLlama-1.1B_aligned_with_baseline_deberta_RM

Base model: TinyLlama/TinyLlama-1.1B-Chat-v1.0

Alignment dataset: openbmb/UltraFeedback

Reward model: payelb/UltraFeedback_openbmb_reward-model-deberta-v3-base_1k_fixed_baseline

Method: PPO alignment with LoRA adapters.

Notes:

  • Reward normalization and clipping enabled
  • KL control enabled
  • pad_token_id/eos_token_id explicitly set
  • DeBERTa RM loaded on a single device (no device_map='auto')