k-diffusion icon indicating copy to clipboard operation
k-diffusion copied to clipboard

Karras samplers: refrain from advancing random generator, allocating unused tensor when gamma==0

Open Birch-san opened this issue 1 year ago • 0 comments

should help with https://github.com/crowsonkb/k-diffusion/issues/25 (giving users more control over determinism), at least for Karras samplers at the default churn=0.

this also eliminates allocation of an unused tensor on every sampler step, which in the case of stable-diffusion was 64kB.

Birch-san avatar Sep 28 '22 00:09 Birch-san