TEA-Lab / diffusion_reward

[ECCV 2024] 💐Official implementation of the paper "Diffusion Reward: Learning Rewards via Conditional Video Diffusion"
https://diffusion-reward.github.io/
MIT License
66 stars 6 forks source link

Pre-trained Reward Models #7

Closed eugeniafeng closed 3 weeks ago

eugeniafeng commented 3 months ago

Hello! I was wondering if it would be possible to release the pre-trained reward models in addition to the codec and video models that are already published (https://huggingface.co/tauhuang/diffusion_reward/tree/main).

TaoHuang13 commented 1 month ago

We use codec and video models as reward models. So downloading these models is equivalent to getting the corresponding reward model.

TaoHuang13 commented 3 weeks ago

Closing because of inactivity.