Prevent upcast to float32 in scheduler

src/euler_scheduler.py

@@ -342,7 +342,7 @@ class MyEulerAncestralDiscreteScheduler(EulerAncestralDiscreteScheduler):
         sigma = self.sigmas[self.step_index]
 
         # Upcast to avoid precision issues when computing prev_sample
-        sample = sample.to(torch.float32)
+        # sample = sample.to(torch.float32)
 
         # 1. compute predicted original sample (x_0) from sigma-scaled predicted noise
         if self.config.prediction_type == "epsilon":