Differenct device in sde_lib.py

[HGGshiwo]

I think line 251 should be

adjacent_sigma = torch.where(timestep == 0, torch.zeros_like(t), 
                                 self.discrete_sigmas.to(t.device)[timestep - 1])

otherwise there accours an error when run reconstruction block in ipynb