Minor fix in ddpm.py

[silky1708]

first move self.logvar to self.device, and then index (line 1030, ddpm.py)

the current ordering gives the following error:

RuntimeError: indices should be either on cpu or on the same device as the indexed tensor (cpu)