search

crowsonkb / k-diffusion

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

time consuming #63

Open fuchao01 opened 1 year ago

fuchao01 commented 1 year ago

image In k-diffusion sampling https://github.com/crowsonkb/k-diffusion/blob/master/k_diffusion/sampling.py#LL56C1-L56C1, it takes 170ms