Merge pull request #1 from silky1708/ddpm

Minor fix in ddpm.py

Author: silky1708

ldm/models/diffusion/ddpm.py

@@ -1027,7 +1027,7 @@ def p_losses(self, x_start, cond, t, noise=None):
         loss_simple = self.get_loss(model_output, target, mean=False).mean([1, 2, 3])
         loss_dict.update({f'{prefix}/loss_simple': loss_simple.mean()})
 
-        logvar_t = self.logvar[t].to(self.device)
+        logvar_t = self.logvar.to(self.device)[t]
         loss = loss_simple / torch.exp(logvar_t) + logvar_t
         # loss = loss_simple / torch.exp(self.logvar) + self.logvar
         if self.learn_logvar: