Choos cuda device - Githubissues

tgxs002 / HPSv2

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

Apache License 2.0

400 stars 12 forks source link

Choos cuda device #23

Open king159 opened 1 year ago

king159 commented 1 year ago

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

tgxs002 commented 1 year ago

Will fix it after the CVPR deadline...