This PR updates the optimizedSD's DDIM sampler to support passing in callbacks via the callback and img_callback args, just like what the PLMS sampler supports. These arguments automatically get passed when calling the DDPM.sample() method, so it now automatically works regardless of the sampler used.
This PR updates the optimizedSD's DDIM sampler to support passing in callbacks via the
callbackandimg_callbackargs, just like what the PLMS sampler supports. These arguments automatically get passed when calling theDDPM.sample()method, so it now automatically works regardless of the sampler used.