From 6e66fec6fbf47f35343b26e945d31daa7e0167b5 Mon Sep 17 00:00:00 2001
From: Artrajz <969242373@qq.com>
Date: Wed, 8 Nov 2023 19:34:48 +0800
Subject: [PATCH] Fix: Add missing torch.cuda.empty_cache() (#161)

---
 infer.py                    | 2 ++
 oldVersion/V101/__init__.py | 2 ++
 2 files changed, 4 insertions(+)

diff --git a/infer.py b/infer.py
index 441e52c..9340ae5 100644
--- 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
diff --git a/oldVersion/V101/__init__.py b/oldVersion/V101/__init__.py
index ddec3b1..cffbed1 100644
--- a/oldVersion/V101/__init__.py
+++ b/oldVersion/V101/__init__.py
@@ -70,4 +70,6 @@ def infer(
             .numpy()
         )
         del x_tst, tones, lang_ids, bert, x_tst_lengths, speakers
+        if torch.cuda.is_available():
+            torch.cuda.empty_cache()
         return audio