From 9cc59ba0891c3ccbebfcbb6784f4dde2d5bd8556 Mon Sep 17 00:00:00 2001
From: UmerHA <40663591+UmerHA@users.noreply.github.com>
Date: Fri, 2 Feb 2024 16:29:00 +0100
Subject: [PATCH] [Contributor Experience] Fix test collection on MPS (#6808)

* Update testing_utils.py

* Update testing_utils.py
---
 src/diffusers/utils/testing_utils.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/src/diffusers/utils/testing_utils.py b/src/diffusers/utils/testing_utils.py
index 86e31eb688..edbf6f31a8 100644
--- a/src/diffusers/utils/testing_utils.py
+++ b/src/diffusers/utils/testing_utils.py
@@ -854,6 +854,8 @@ def _is_torch_fp64_available(device):
 
     import torch
 
+    device = torch.device(device)
+
     try:
         x = torch.zeros((2, 2), dtype=torch.float64).to(device)
         _ = torch.mul(x, x)