Hi, I encountered a bug when running the real image sampling script.
File "xxxxx/score-MRI/sde_lib.py", line 156, in discretize
self.discrete_sigmas[timestep - 1].to(t.device))
RuntimeError: indices should be either on cpu or on the same device as the indexed tensor (cpu)
Hi, I encountered a bug when running the real image sampling script.
a simple fix can solve this problem