fix(randn_tensor): compare device.type, not torch.device, when suppressing MPS info log

src/diffusers/utils/torch_utils.py

@@ -173,7 +173,7 @@ def randn_tensor(
         gen_device_type = generator.device.type if not isinstance(generator, list) else generator[0].device.type
         if gen_device_type != device.type and gen_device_type == "cpu":
             rand_device = "cpu"
-            if device != "mps":
+            if device.type != "mps":
                 logger.info(
                     f"The passed generator was created on 'cpu' even though a tensor on {device} was expected."
                     f" Tensors will be created on 'cpu' and then moved to {device}. Note that one can probably"

tests/others/test_utils.py

@@ -247,6 +247,43 @@ def test_fourier_filter_preserves_dtype_and_shape(self):
             assert out.shape == x.shape
 
 
+class RandnTensorTester(unittest.TestCase):
+    """Tests for :func:`diffusers.utils.torch_utils.randn_tensor`."""
+
+    def test_mps_suppresses_cpu_generator_info_log(self):
+        import torch
+
+        from diffusers.utils import logging as diffusers_logging
+        from diffusers.utils import torch_utils
+
+        from ..testing_utils import CaptureLogger
+
+        gen = torch.Generator(device="cpu")
+        diffusers_logging.set_verbosity_info()
+
+        def _capture(target_device):
+            with CaptureLogger(torch_utils.logger) as cl:
+                try:
+                    torch_utils.randn_tensor((1, 2), generator=gen, device=target_device, dtype=torch.float32)
+                except (AssertionError, RuntimeError):
+                    pass
+            return cl.out
+
+        mps_out = _capture("mps")
+        self.assertNotIn(
+            "moved to",
+            mps_out,
+            f"MPS target should not emit the CPU-fallback info log, got: {mps_out}",
+        )
+
+        cuda_out = _capture("cuda")
+        self.assertIn(
+            "moved to",
+            cuda_out,
+            f"Non-MPS target should still emit the CPU-fallback info log, got: {cuda_out}",
+        )
+
+
 # Copied from https://github.com/huggingface/transformers/blob/main/tests/utils/test_expectations.py
 class ExpectationsTester(unittest.TestCase):
     def test_expectations(self):