From 5adb7a11bcc70d26413ba3e8627ca72d99b01840 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Stardust=C2=B7=E5=87=8F?= Date: Tue, 5 Sep 2023 20:39:56 +0800 Subject: [PATCH] add empty cache --- webui.py | 1 + 1 file changed, 1 insertion(+) diff --git a/webui.py b/webui.py index edbf03f..130ba9f 100644 --- a/webui.py +++ b/webui.py @@ -78,6 +78,7 @@ def infer(text, sdp_ratio, noise_scale, noise_scale_w, length_scale, sid, langua def tts_fn(text, speaker, sdp_ratio, noise_scale, noise_scale_w, length_scale, language): with torch.no_grad(): audio = infer(text, sdp_ratio=sdp_ratio, noise_scale=noise_scale, noise_scale_w=noise_scale_w, length_scale=length_scale, sid=speaker, language=language) + torch.cuda.empty_cache() return "Success", (hps.data.sampling_rate, audio)