oliversssf2 commited on
Commit
aa3e1bf
1 Parent(s): 0149ae6

Create README.md

Browse files
Files changed (1) hide show
  1. README.md +6 -0
README.md ADDED
@@ -0,0 +1,6 @@
 
 
 
 
 
 
 
1
+ Load the reward model with the following code
2
+
3
+ ```python
4
+ from reward_modelling.reward_model import RewardModel
5
+ rm = RewardModel.from_pretrained('oliversssf2/distilbert-base-uncased-rm-helpful')
6
+ ```