File size: 409 Bytes
19c955a |
1 2 3 4 5 6 7 8 9 10 11 12 13 |
## DDPO aesthetic predictor
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.
```python
import torch
from huggingface_hub import hf_hub_download
cached_path = hf_hub_download(
'trl-lib',
'aesthetic-model.pth'
)
``` |