Spaces:

mattmdjaga
/

Preference_Scorer-Pickscore

Runtime error

mattmdjaga commited on Feb 4, 2024

Commit

9aff4b8

verified ·

1 Parent(s): 13e3eda

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -13,7 +13,7 @@ device = "cuda" if torch.cuda.is_available() else "cpu"
 dtype = torch.float16 if device == "cuda" else torch.float32
 # Load example dataset
-dataset = load_dataset("xzuyn/dalle-3_vs_sd-v1-5_dpo", num_proc=8)
 processor_name = "laion/CLIP-ViT-H-14-laion2B-s32B-b79K"
 model_name = "yuvalkirstain/PickScore_v1"

 dtype = torch.float16 if device == "cuda" else torch.float32
 # Load example dataset
+dataset = load_dataset("xzuyn/dalle-3_vs_sd-v1-5_dpo", num_proc=4)
 processor_name = "laion/CLIP-ViT-H-14-laion2B-s32B-b79K"
 model_name = "yuvalkirstain/PickScore_v1"