This library is not required to use k-diffusion Make k-diffusion wrapper closer to the other samplers