stable-diffusion
stable-diffusion copied to clipboard
Implement callback support for DDIM sampler
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.