Closed colinjneville closed 1 year ago
I can confirm this bug.
Also, there are many other divisions that in the python library are div_floor divisions that can potentially create similar problems.
I guess we can either do some rounding to help with this or at least check that the numbers are divisible and if not make a nicer error message so that the underlying issue is easier to diagnose.
I just pushed a very simple fix for this, feel free to re-open if you run into further issues.
DDIMScheduler
's first time step is effectivelyinference_steps * train_timesteps / inference_steps
, which if there is no truncation, puts it exactly attrain_timesteps
, just out of bounds.