get_vc bug fix

2023-04-06 11:31:20 +08:00 · 2023-04-06 11:31:20 +08:00 · 6062b95b0e
parent 86f76f8a03
commit 6062b95b0e
1 changed files with 14 additions and 13 deletions
--- a/infer-web.py
+++ b/infer-web.py
@ -148,8 +148,9 @@ def get_vc(sid):
    global n_spk,tgt_sr,net_g,vc,cpt
    if(sid==[]):
        global hubert_model
+        if (hubert_model != None): # 考虑到轮询，需要加个判断看是否 sid 是由有模型切换到无模型的
            print("clean_empty_cache")
-        del n_spk, vc, hubert_model,tgt_sr#,cpt
+            del net_g, n_spk, vc, hubert_model,tgt_sr#,cpt
            hubert_model = net_g=n_spk=vc=hubert_model=tgt_sr=None
            torch.cuda.empty_cache()
            ###楼下不这么折腾清理不干净