kaiwang960112 / SpeeD

SpeeD: A Closer Look at Time Steps is Worthy of Triple Speed-Up for Diffusion Model Training
Apache License 2.0
132 stars 3 forks source link

Does the method work with other beta schedules? #5

Open DianCh opened 2 weeks ago

DianCh commented 2 weeks ago

Hi authors, thank you for releasing this nice work. The paper's findings seemed to be based on the linear beta schedule, and I'm wondering if the method would also work with other beta schedules, like sigmoid or cosine?

1zeryu commented 1 week ago

Thank you for your question, we try other beta schedules and report the results in section 3.3 cross-architecture robustness evaluation. Our method works on different beta schedule like 'linear', 'cosine', 'quadratic'. If there are any other questions, feel free to contact us.