| ## 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' | |
| ) | |
| state_dict = torch.load(cached_path) | |
| ``` |