oliversssf2's picture
Create README.md
aa3e1bf
|
Raw
History Blame Contribute Delete
197 Bytes

Load the reward model with the following code

from reward_modelling.reward_model import RewardModel
rm = RewardModel.from_pretrained('oliversssf2/distilbert-base-uncased-rm-helpful')