diffusers
0343d8f5 - Do not use torch.float64 on the mps device (#942)

Commit

2 years ago

Do not use torch.float64 on the mps device (#942) * Add failing test for #940. * Do not use torch.float64 in mps. * style * Temporarily skip add_noise for IPNDMScheduler. Until #990 is addressed.

References

#942 - Do not use torch.float64 on the mps device

Author

pcuenca

Parents

4b9f5895

Files2

src/diffusers/schedulers
- scheduling_lms_discrete.py
tests
- test_scheduler.py

diffusers 0343d8f5 - Do not use torch.float64 on the mps device (#942)

diffusers
0343d8f5 - Do not use torch.float64 on the mps device (#942)