Thanks for your great work! I wonder if this repo is able to train with hybrid loss ($\lambda L_{vlb} + L_{simple}$) proposed in improved DDPM？
