Giters

tgxs002 / HPSv2

Human Preference Score v2: A Solid Benchmark for Evaluating Human Preferences of Text-to-Image Synthesis

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Choos cuda device

king159 opened this issue · comments

commented

It will be helpful if we can pass in the torch device in the hpsv2.score(..., device=torch.device("cuda") function. Currently, the model will be forced to run on GPU0

commented

Will fix it after the CVPR deadline...