trl-lib
/

ddpo-aesthetic-predictor

Model card Files Files and versions

ybelkada commited on Aug 18, 2023

Commit

cf4bd10

·

1 Parent(s): 7f63969

Update README.md

Files changed (1) hide show

README.md +3 -1

README.md CHANGED Viewed

@@ -9,5 +9,7 @@ from huggingface_hub import hf_hub_download
 cached_path = hf_hub_download(
   'trl-lib',
   'aesthetic-model.pth'
-)
 ```

 cached_path = hf_hub_download(
   'trl-lib',
   'aesthetic-model.pth'
+)
+state_dict = torch.load(cached_path)
 ```