update code

Signed-off-by: Liu, Kaixuan <[email protected]>
huggingface · Dec 9, 2024 · b6a31c9 · b6a31c9
1 parent 1e404bb
commit b6a31c9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/ipex/test_pipelines.py b/tests/ipex/test_pipelines.py
@@ -35,7 +35,7 @@
 
 
 torch.use_deterministic_algorithms(True)
-DEVICE = "xpu:0" if IS_XPU_AVAILABLE else None
+DEVICE = "xpu:0" if IS_XPU_AVAILABLE else "cpu"
 
 
 class PipelinesIntegrationTest(unittest.TestCase):