Upload myinfer_latest.py

myinfer_latest.py

@@ -211,7 +211,7 @@ def get_processed_audio(audio_id):
 
 
 def convert_voice(spk_id, input_audio_path, voice_transform,unique_id):
-    get_vc_safe(spk_id,0.5)
+    get_vc(spk_id,0.5)
     print("*****before makinf call to vc ", unique_id)
 
    
@@ -380,22 +380,22 @@ def get_vc(sid, to_return_protect0):
             if torch.cuda.is_available():
                 torch.cuda.empty_cache()
             ###楼下不这么折腾清理不干净
-            if_f0 = cpt.get("f0", 1)
-            version = cpt.get("version", "v1")
+            if_f0 = cpt[sid].get("f0", 1)
+            version = cpt[sid].get("version", "v1")
             if version == "v1":
                 if if_f0 == 1:
                     net_g = SynthesizerTrnMs256NSFsid(
-                        *cpt["config"], is_half=config.is_half
+                        *cpt[sid]["config"], is_half=config.is_half
                     )
                 else:
-                    net_g = SynthesizerTrnMs256NSFsid_nono(*cpt["config"])
+                    net_g = SynthesizerTrnMs256NSFsid_nono(*cpt[sid]["config"])
             elif version == "v2":
                 if if_f0 == 1:
                     net_g = SynthesizerTrnMs768NSFsid(
-                        *cpt["config"], is_half=config.is_half
+                        *cpt[sid]["config"], is_half=config.is_half
                     )
                 else:
-                    net_g = SynthesizerTrnMs768NSFsid_nono(*cpt["config"])
+                    net_g = SynthesizerTrnMs768NSFsid_nono(*cpt[sid]["config"])
             del net_g, cpt
             if torch.cuda.is_available():
                 torch.cuda.empty_cache()
@@ -408,10 +408,10 @@ def get_vc(sid, to_return_protect0):
         )
     print(f"Loading {sid} model...")
     selected_model = sid[:-4]
-    cpt = torch.load(os.path.join(weight_root, sid), map_location="cpu")
-    tgt_sr = cpt["config"][-1]
-    cpt["config"][-3] = cpt["weight"]["emb_g.weight"].shape[0]
-    if_f0 = cpt.get("f0", 1)
+    cpt[sid] = torch.load(os.path.join(weight_root, sid), map_location="cpu")
+    tgt_sr = cpt[sid]["config"][-1]
+    cpt[sid]["config"][-3] = cpt[sid]["weight"]["emb_g.weight"].shape[0]
+    if_f0 = cpt[sid].get("f0", 1)
     if if_f0 == 0:
         to_return_protect0 = {
             "visible": False,
@@ -424,26 +424,26 @@ def get_vc(sid, to_return_protect0):
             "value": to_return_protect0,
             "__type__": "update",
         }
-    version = cpt.get("version", "v1")
+    version = cpt[sid].get("version", "v1")
     if version == "v1":
         if if_f0 == 1:
-            net_g = SynthesizerTrnMs256NSFsid(*cpt["config"], is_half=config.is_half)
+            net_g = SynthesizerTrnMs256NSFsid(*cpt[sid]["config"], is_half=config.is_half)
         else:
-            net_g = SynthesizerTrnMs256NSFsid_nono(*cpt["config"])
+            net_g = SynthesizerTrnMs256NSFsid_nono(*cpt[sid]["config"])
     elif version == "v2":
         if if_f0 == 1:
-            net_g = SynthesizerTrnMs768NSFsid(*cpt["config"], is_half=config.is_half)
+            net_g = SynthesizerTrnMs768NSFsid(*cpt[sid]["config"], is_half=config.is_half)
         else:
-            net_g = SynthesizerTrnMs768NSFsid_nono(*cpt["config"])
+            net_g = SynthesizerTrnMs768NSFsid_nono(*cpt[sid]["config"])
     del net_g.enc_q
-    print(net_g.load_state_dict(cpt["weight"], strict=False))
+    print(net_g.load_state_dict(cpt[sid]["weight"], strict=False))
     net_g.eval().to(config.device)
     if config.is_half:
         net_g = net_g.half()
     else:
         net_g = net_g.float()
     vc = VC(tgt_sr, config)
-    n_spk = cpt["config"][-3]
+    n_spk = cpt[sid]["config"][-3]
     weights_index = []
     for _, _, index_files in os.walk(index_root):
         for file in index_files: