blackteay/GPT-SoVITS
1
0
Fork 0
mirror of https://github.com/RVC-Boss/GPT-SoVITS.git synced 2025-04-06 03:57:44 +08:00
Code Issues Packages Projects Releases Wiki Activity

优化代码

Browse Source
This commit is contained in:
chasonjiang 2024-03-14 11:15:09 +08:00
parent 0ff60a947a
commit 8698c28c90
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
GPT_SoVITS/TTS_infer_pack/TTS.py
View File

@ -820,7 +820,7 @@ class TTS:
def empty_cache(self): def empty_cache(self):
try: try:
if str(self.configs.device) == "cuda": if "cuda" in str(self.configs.device):
torch.cuda.empty_cache() torch.cuda.empty_cache()
elif str(self.configs.device) == "mps": elif str(self.configs.device) == "mps":
torch.mps.empty_cache() torch.mps.empty_cache()