Update ldm/models/diffusion/ddim.py

ldm/models/diffusion/ddim.py

@@ -16,7 +16,7 @@ class DDIMSampler(object):
         self.schedule = schedule
 
     def register_buffer(self, name, attr):
-        if type(attr) == torch.Tensor:
+        # if type(attr) == torch.Tensor:
             # if attr.device != torch.device("cuda"):
             #     attr = attr.to(torch.device("cuda"))
         setattr(self, name, attr)