Fix: server_fastapi.py, support auto split (#152)

* Fix: server_fastapi.py, support auto split * Fix: translate.py
2023-11-06 19:40:58 +08:00
parent 855774213e
commit 096d661a4b
3 changed files with 45 additions and 20 deletions
--- a/infer.py
+++ b/infer.py
@@ -201,6 +201,5 @@ def infer(
            .float()
            .numpy()
        )
-        del x_tst, tones, lang_ids, bert, x_tst_lengths, speakers
-        torch.cuda.empty_cache()
+        del x_tst, tones, lang_ids, bert, x_tst_lengths, speakers, ja_bert, en_bert
        return audio