search

Anima-Lab / MaskDiT

Code for Fast Training of Diffusion Models with Masked Transformers
MIT License
377 stars 14 forks source link

Diffusion Schedule #12

Open yangliuxin-nn opened 5 months ago

yangliuxin-nn commented 5 months ago

Hi, I am writing to ask which diffusion schedule do you use, linear or cosine? Many thanks for your time!

devzhk commented 5 months ago

Hi, We didn't use DDPM framework. We follow the EDM schedule, which is a Variance-Exploding scheme. You can check out the EDM scheme in Table 1 of EDM paper