search

crowsonkb / k-diffusion

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

add gradient accumulation support #12

Closed tmabraham closed 2 years ago

tmabraham commented 2 years ago

use Accelerate's API for this, add a new hyperparameter to the configs...

crowsonkb commented 2 years ago

I changed some small things then merged this PR: https://github.com/crowsonkb/k-diffusion/commit/d6e087eedea5df1cd387a633d981a403de1fc68e

Thank you! :)