prevent replacing torch_randn globally (instead replacing k_diffusion.sampling.torch) and add a setting to disable this all