optimize(infer): move syns into rvc

Author: fumiama

infer/lib/jit/synthesizer.py

@@ -1,35 +1,20 @@
 import torch
 
+from rvc.synthesizers import SynthesizerTrnMsNSFsid
 
-def get_synthesizer_ckpt(cpt, device=torch.device("cpu")):
-    from infer.lib.infer_pack.models import (
-        SynthesizerTrnMs256NSFsid,
-        SynthesizerTrnMs256NSFsid_nono,
-        SynthesizerTrnMs768NSFsid,
-        SynthesizerTrnMs768NSFsid_nono,
-    )
 
-    # tgt_sr = cpt["config"][-1]
+def get_synthesizer_ckpt(cpt, device=torch.device("cpu")):
     cpt["config"][-3] = cpt["weight"]["emb_g.weight"].shape[0]
     if_f0 = cpt.get("f0", 1)
     version = cpt.get("version", "v1")
     if version == "v1":
-        if if_f0 == 1:
-            net_g = SynthesizerTrnMs256NSFsid(*cpt["config"])
-        else:
-            net_g = SynthesizerTrnMs256NSFsid_nono(*cpt["config"])
+        encoder_dim = 256
     elif version == "v2":
-        if if_f0 == 1:
-            net_g = SynthesizerTrnMs768NSFsid(*cpt["config"])
-        else:
-            net_g = SynthesizerTrnMs768NSFsid_nono(*cpt["config"])
+        encoder_dim = 768
+    net_g = SynthesizerTrnMsNSFsid(
+        *cpt["config"], encoder_dim=encoder_dim, use_f0 = if_f0==1,
+    )
     del net_g.enc_q
-    # net_g.forward = net_g.infer
-    # ckpt = {}
-    # ckpt["config"] = cpt["config"]
-    # ckpt["f0"] = if_f0
-    # ckpt["version"] = version
-    # ckpt["info"] = cpt.get("info", "0epoch")
     net_g.load_state_dict(cpt["weight"], strict=False)
     net_g = net_g.float()
     net_g.eval().to(device)

infer/modules/train/train.py

@@ -59,12 +59,12 @@
 from rvc.discriminators import MultiPeriodDiscriminator
 
 if hps.version == "v1":
-    from infer.lib.infer_pack.models import SynthesizerTrnMs256NSFsid as RVC_Model_f0
-    from infer.lib.infer_pack.models import (
+    from rvc.synthesizers import SynthesizerTrnMs256NSFsid as RVC_Model_f0
+    from rvc.synthesizers import (
         SynthesizerTrnMs256NSFsid_nono as RVC_Model_nof0,
     )
 else:
-    from infer.lib.infer_pack.models import (
+    from rvc.synthesizers import (
         SynthesizerTrnMs768NSFsid as RVC_Model_f0,
         SynthesizerTrnMs768NSFsid_nono as RVC_Model_nof0,
     )

infer/lib/infer_pack/models.py renamed to rvc/synthesizers.py

@@ -24,7 +24,7 @@
 
 # from models import SynthesizerTrn256#hifigan_nonsf
 # from lib.infer_pack.models import SynthesizerTrn256NSF as SynthesizerTrn256#hifigan_nsf
-from infer.lib.infer_pack.models import (
+from rvc.synthesizers import (
     SynthesizerTrnMs256NSFsid as SynthesizerTrn256,
 )  # hifigan_nsf
 from scipy.io import wavfile