You can deploy this model and start earning money today!

trl-lib/
ddpo-aesthetic-predictor

Name: ddpo-aesthetic-predictor
Availability: InStock

Views

No views yet

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)

trl-lib/ddpo-aesthetic-predictor

DDPO aesthetic predictor

trl-lib/
ddpo-aesthetic-predictor