blackteay/GPT-SoVITS
1
0
Fork 0
mirror of https://github.com/RVC-Boss/GPT-SoVITS.git synced 2025-06-10 02:59:17 +08:00
Code Issues Packages Projects Releases Wiki Activity

Update api.py

Browse Source
This commit is contained in:
YongKang Zhou 2024-01-31 10:47:13 +08:00 committed by GitHub
parent 53817b983c
commit 956ae58677
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
api.py
View File

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