Update api.py

2 years ago · 956ae58677
parent 53817b983c
commit 956ae58677
1 changed files with 2 additions and 1 deletions
--- a/api.py
+++ b/api.py
@ -439,7 +439,8 @@ def handle(refer_wav_path, prompt_text, prompt_language, text, text_language):
    wav.seek(0)
    torch.cuda.empty_cache()
-    torch.mps.empty_cache()
+    if device == "mps":
        torch.mps.empty_cache()
    return StreamingResponse(wav, media_type="audio/wav")