Fix: Add missing torch.cuda.empty_cache() (#161)

2023-11-08 19:34:48 +08:00
parent 6a77c8ba1d
commit 6e66fec6fb
2 changed files with 4 additions and 0 deletions
--- a/infer.py
+++ b/infer.py
@@ -202,4 +202,6 @@ def infer(
            .numpy()
        )
        del x_tst, tones, lang_ids, bert, x_tst_lengths, speakers, ja_bert, en_bert
+        if torch.cuda.is_available():
+            torch.cuda.empty_cache()
        return audio