tuna2134/sbv2-v2
1
0
Fork 0
Code Issues Pull Requests 1 Actions Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
Artrajz
2023-11-08 19:34:48 +08:00
committed by GitHub
parent 6a77c8ba1d
commit 6e66fec6fb
2 changed files with 4 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
infer.py
View File

@@ -202,4 +202,6 @@ def infer(
.numpy() .numpy()
) )
del x_tst, tones, lang_ids, bert, x_tst_lengths, speakers, ja_bert, en_bert 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 return audio

2
oldVersion/V101/__init__.py
View File

@@ -70,4 +70,6 @@ def infer(
.numpy() .numpy()
) )
del x_tst, tones, lang_ids, bert, x_tst_lengths, speakers del x_tst, tones, lang_ids, bert, x_tst_lengths, speakers
if torch.cuda.is_available():
torch.cuda.empty_cache()
return audio return audio