fix memory leak in streamlit app

manmay-nakhashi · Jul 29, 2023 · 6b8caa8 · 6b8caa8
1 parent ec420c1
commit 6b8caa8
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/tortoise/inference.py b/tortoise/inference.py
@@ -102,13 +102,13 @@ def run_and_save_tts(
     #
     if not isinstance(gen, list):
         gen = [gen]
-    gen = [g.squeeze(0).cpu() for g in gen]
+    audio_ = [g.squeeze(0).cpu() for g in gen]
     fps = []
-    for i, g in enumerate(gen):
+    for i, g in enumerate(audio_):
         fps.append(output_dir / f"{i}.wav")
         save_gen_with_voicefix(g, fps[-1], squeeze=False, voicefixer=voicefixer)
         # torchaudio.save(output_dir/f'{i}.wav', g, 24000)
-    return fps if return_filepaths else gen
+    return fps if return_filepaths else audio_
 
 
 def infer_on_texts(