openai / guided-diffusion

MIT License
6.11k stars 813 forks source link

Scheduler does not work with any number of diffusion time steps #137

Open nilsleh opened 8 months ago

nilsleh commented 8 months ago

The Linear Schedule suggests that any number of time-steps are supported. But this is not true, since the Guided Diffusion asserts that all betas are in the interval (0,1). So any num_diffusion_timesteps<200 will yield an assertion error.

To reproduce, train any model with the flag diffusion_steps < 200.