Closed tmabraham closed 2 years ago
use Accelerate's API for this, add a new hyperparameter to the configs...
I changed some small things then merged this PR: https://github.com/crowsonkb/k-diffusion/commit/d6e087eedea5df1cd387a633d981a403de1fc68e
Thank you! :)
use Accelerate's API for this, add a new hyperparameter to the configs...