fix flaky cpu test

tests/test_modeling_utils.py

@@ -1046,8 +1046,12 @@ class PipelineTesterMixin(unittest.TestCase):
         torch.manual_seed(0)
         image = sde_ve(num_inference_steps=2)
 
-        expected_image_sum = 3382849024.0
-        expected_image_mean = 1075.3788
+        if model.device.type == "cpu":
+            expected_image_sum = 3384805888.0
+            expected_image_mean = 1076.00085
+        else:
+            expected_image_sum = 3382849024.0
+            expected_image_mean = 1075.3788
 
         assert (image.abs().sum() - expected_image_sum).abs().cpu().item() < 1e-2
         assert (image.abs().mean() - expected_image_mean).abs().cpu().item() < 1e-4