This commit is contained in:
Joseph Cheek 2025-08-04 09:35:50 -07:00 committed by GitHub
commit 7c159c6520
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -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)