public-mirrors/Kokoro-FastAPI
1
0
Fork 0
mirror of https://github.com/remsky/Kokoro-FastAPI.git synced 2025-08-05 16:48:53 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Fix test_initial_state() test failure on apple silicon.

Browse source
This commit is contained in:
Joseph Cheek 2025-07-26 17:06:37 -06:00
parent 29066f7c9f
commit c16ff335e4
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
api/tests/test_kokoro_v1.py
View file

@ -19,7 +19,7 @@ def test_initial_state(kokoro_backend):
assert kokoro_backend._model is None assert kokoro_backend._model is None
assert kokoro_backend._pipelines == {} # Now using dict of pipelines assert kokoro_backend._pipelines == {} # Now using dict of pipelines
# Device should be set based on settings # 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) @patch("torch.cuda.is_available", return_value=True)