search

PKU-YuanGroup / Open-Sora-Plan

This project aim to reproduce Sora (Open AI T2V model), we wish the open source community contribute to this project.
Apache License 2.0
10.87k stars 971 forks source link

add support snr_gamma for rebalancing loss #279

Open Pang-Yatian opened 1 month ago

Pang-Yatian commented 1 month ago

Add support snr_gamma for rebalancing loss.

Reference https://arxiv.org/abs/2303.09556