Downloads · 30 days
0
ddpo-aesthetic-predictor
trl-lib/ddpo-aesthetic-predictor
ddpo-aesthetic-predictor is a machine learning model from trl-lib. Use it for the machine learning task on the model card, and read the license before you ship it in a product.
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 hugging…
Access
Public
Updated Aug 18, 2023
Repo size
3.7 MB
Likes
4
Public
Repo makeup
Click a slice to open those files.
.pth3.7 MB · 100%
At a glance
- Access
- Public
- Created
- Aug 18, 2023
- Updated
- Aug 18, 2023
- SHA
- cf4bd103
Model card
From the Hugging Face model README
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.
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)
- Created
- Aug 18, 2023
- Updated
- Aug 18, 2023