add mps empty cache

faaany · web-flow · commit d820f75c16d6 · 2025-01-08T17:46:37.000+08:00
diff --git a/src/diffusers/utils/testing_utils.py b/src/diffusers/utils/testing_utils.py
@@ -1067,7 +1067,7 @@ def _is_torch_fp64_available(device):
         "cuda": torch.cuda.empty_cache,
         "xpu": torch.xpu.empty_cache,
         "cpu": None,
-        "mps": None,
+        "mps": torch.mps.empty_cache,
         "default": None,
     }
     BACKEND_DEVICE_COUNT = {

Original file line number	Diff line number	Diff line change
`@@ -1067,7 +1067,7 @@ def _is_torch_fp64_available(device):`
`1067`	`1067`	`"cuda": torch.cuda.empty_cache,`
`1068`	`1068`	`"xpu": torch.xpu.empty_cache,`
`1069`	`1069`	`"cpu": None,`
`1070`		`- "mps": None,`
	`1070`	`+ "mps": torch.mps.empty_cache,`
`1071`	`1071`	`"default": None,`
`1072`	`1072`	`}`
`1073`	`1073`	`BACKEND_DEVICE_COUNT = {`