fix float32 DPS

Author: tachella

deepinv_bench/benchmarks/div2k_gaussian_deblurring/solvers/dps.py

@@ -30,7 +30,7 @@ def set_objective(self, train_dataset=None, physics=None):
             raise NotImplementedError
 
         self.model = dinv.sampling.DPS(
-            model=denoiser, device=device, max_iter=self.max_iter
+            model=denoiser, device=device, max_iter=self.max_iter, dtype=torch.float32
         )
         self.model.device = device
 

deepinv_bench/benchmarks/div2k_inpainting_easy/solvers/dps.py

@@ -30,7 +30,7 @@ def set_objective(self, train_dataset=None, physics=None):
             raise NotImplementedError
 
         self.model = dinv.sampling.DPS(
-            model=denoiser, device=device, max_iter=self.max_iter
+            model=denoiser, device=device, max_iter=self.max_iter, dtype=torch.float32
         )
         self.model.device = device
 

deepinv_bench/benchmarks/div2k_super_resolution_2x/solvers/dps.py

@@ -30,7 +30,7 @@ def set_objective(self, train_dataset=None, physics=None):
             raise NotImplementedError
 
         self.model = dinv.sampling.DPS(
-            model=denoiser, device=device, max_iter=self.max_iter
+            model=denoiser, device=device, max_iter=self.max_iter, dtype=torch.float32
         )
         self.model.device = device