torch diffusers numpy imageio