From c16ff335e46443faede7769b36acbfeedf88a256 Mon Sep 17 00:00:00 2001 From: Joseph Cheek Date: Sat, 26 Jul 2025 17:06:37 -0600 Subject: [PATCH] Fix test_initial_state() test failure on apple silicon. --- api/tests/test_kokoro_v1.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/api/tests/test_kokoro_v1.py b/api/tests/test_kokoro_v1.py index 29d83c5..6891d8d 100644 --- a/api/tests/test_kokoro_v1.py +++ b/api/tests/test_kokoro_v1.py @@ -19,7 +19,7 @@ def test_initial_state(kokoro_backend): assert kokoro_backend._model is None assert kokoro_backend._pipelines == {} # Now using dict of pipelines # Device should be set based on settings - assert kokoro_backend.device in ["cuda", "cpu"] + assert kokoro_backend.device in ["cuda", "cpu", "mps"] @patch("torch.cuda.is_available", return_value=True)