优化代码

2024-03-14 11:15:09 +08:00
parent 0ff60a947a
commit 8698c28c90
1 changed files with 1 additions and 1 deletions
--- a/GPT_SoVITS/TTS_infer_pack/TTS.py
+++ b/GPT_SoVITS/TTS_infer_pack/TTS.py
@@ -820,7 +820,7 @@ class TTS:
    
    def empty_cache(self):
        try:    
-            if str(self.configs.device) == "cuda":
+            if "cuda" in str(self.configs.device):
                torch.cuda.empty_cache()
            elif str(self.configs.device) == "mps":
                torch.mps.empty_cache()