bacow/GPT-SoVITS
1
0
Fork 0
Code Issues Pull Requests Actions 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
1 changed files with 2 additions and 1 deletions
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) wav.seek(0)
torch.cuda.empty_cache() torch.cuda.empty_cache()
torch.mps.empty_cache() if device == "mps":
torch.mps.empty_cache()
return StreamingResponse(wav, media_type="audio/wav") return StreamingResponse(wav, media_type="audio/wav")