ybelkada commited on
Commit
19c955a
1 Parent(s): edfc0b1

Create README.md

Browse files
Files changed (1) hide show
  1. README.md +13 -0
README.md ADDED
@@ -0,0 +1,13 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ## DDPO aesthetic predictor
2
+
3
+ This reprository contains the weights of the aesthetic predictor that you can find in the repository: https://github.com/christophschuhmann/improved-aesthetic-predictor so that any use can load it easily using `huggingface_hub` library.
4
+
5
+ ```python
6
+ import torch
7
+ from huggingface_hub import hf_hub_download
8
+
9
+ cached_path = hf_hub_download(
10
+ 'trl-lib',
11
+ 'aesthetic-model.pth'
12
+ )
13
+ ```