Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Fix img2img speed with LMS-Discrete Scheduler (huggingface#896)
Casting `self.sigmas` into a different dtype (the one of original_samples) is not advisable. In my img2img pipeline this leads to a long running time in the `integrate.quad` call later on- by long I mean more than 10x slower. Co-authored-by: Anton Lozhkov <anton@huggingface.co>
- Loading branch information