Update models/gaussian_diffusion.py

models/gaussian_diffusion.py

@@ -29,6 +29,7 @@ class DiffusePipeline(object):
         self.num_inference_steps = num_inference_steps
         if self.torch_dtype == torch.float16:
             model = model.half()
+        print("Xxx",device,"xxxx")
         self.model = model.to(device)
         self.opt = opt