trl icon indicating copy to clipboard operation
trl copied to clipboard

How to Add PEFT to PPO Trainer or PPO Config

Open ZhichaoWang970201 opened this issue 1 year ago • 1 comments

I am trying to realize RLHF through PPO.

May I ask how can I realize PEFT in RLHF/PPO. I can see this parameter in DPOTrainer. However, I cannot see that in PPOTrainer.

ZhichaoWang970201 avatar Aug 12 '24 01:08 ZhichaoWang970201

PPO does not yet support PEFT. But it may be a good enhancement.

qgallouedec avatar Oct 07 '24 14:10 qgallouedec

Hi @qgallouedec, I'd love to contribute to this enhancement! I noticed that this issue has been open for three months, and I'd like to help bring it forward.

Is there anything specific I should be mindful of as I start on this? Any guidance or notes on the current status would be greatly appreciated.

Thank you!

ccs96307 avatar Nov 08 '24 16:11 ccs96307