ImageReward icon indicating copy to clipboard operation
ImageReward copied to clipboard

Published 20 hours ago • verified publisher icon THUDM

Finetuned model weights

Open rohitrango opened this issue 1 year ago • 4 comments

Hi,

Thank you for releasing code for the work! Have you released the weights for any finetuned model in the paper?

Thanks.

rohitrango avatar Aug 05 '24 22:08 rohitrango

Thanks for your attention! You can construct your own prompt set to get a model that better meets your needs.

#84

xujz18 avatar Aug 06 '24 02:08 xujz18

Why does the script scripts/train_refl_sdxl.sh require images too (train_refl.sh doesnt)?

rohitrango avatar Aug 07 '24 20:08 rohitrango

Why does the script scripts/train_refl_sdxl.sh require images too (train_refl.sh doesnt)?

I meet the same question, do you have solved it? i don't understand what the images refer to..

Cominclip avatar Aug 14 '24 11:08 Cominclip

This is code contributed by a community friend who supposedly added the pre-training loss function mentioned in the paper. You can refer to https://github.com/THUDM/ImageReward/pull/89

xujz18 avatar Aug 18 '24 02:08 xujz18