search

crowsonkb / k-diffusion

Karras et al. (2022) diffusion models for PyTorch
MIT License
2.21k stars 371 forks source link

bugfix: mps device precision error #77

Closed afpro closed 11 months ago

afpro commented 11 months ago
  1. x.log() and x.exp with small x return obvious wrong result
  2. x[..., None, None] sometimes return just 0
afpro commented 11 months ago

this should not be fixed in this way, wait pytorch bugfix.