add models

app.py

@@ -185,8 +185,11 @@ if __name__ == '__main__':
             **hps.model)
         utils.load_checkpoint(model_path, model, None)
         model.eval().to(device)
-        speaker_ids = [sid for sid, name in enumerate(hps.speakers) if name != "None"]
-        speakers = [name for sid, name in enumerate(hps.speakers) if name != "None"]
+        if isinstance(hps.speakers, utils.HParams):
+            speakers, speaker_ids = zip(*hps.speakers.items())
+        else:
+            speaker_ids = [sid for sid, name in enumerate(hps.speakers) if name != "None"]
+            speakers = [name for sid, name in enumerate(hps.speakers) if name != "None"]
 
         t = info["type"]
         if t == "vits":

export_model.py

@@ -1,8 +1,8 @@
 import torch
 
 if __name__ == '__main__':
-    model_path = "saved_model/18/model.pth"
-    output_path = "saved_model/18/model1.pth"
+    model_path = "saved_model/19/model.pth"
+    output_path = "saved_model/19/model1.pth"
     checkpoint_dict = torch.load(model_path, map_location='cpu')
     checkpoint_dict_new = {}
     for k, v in checkpoint_dict.items():

saved_model/19/config.json

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:2b58c83def56c61fb64a937fe4485f0a35817abc5df8332cb58fc30cbba19c76
+size 8812

saved_model/19/model.pth

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:5e250a75c0fb446825edcfba0b1ebc0872afff50143d66f3327a3eeb628bd874
+size 159065393

saved_model/20/config.json

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:540163625e921ccbbf435e96affff02f99b36f9ffcaa47e57b01031a7721b696
+size 3572

saved_model/20/model.pth

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:4eebab160b105e0c024d2291715eb38063d8fad0b26fa76624e84e0e7057f8f5
+size 159706189

saved_model/info.json

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:ec1565a8f12e8d333dc8d233a472cc24398722f2fe2b63dd8223a5fea18ef06d
-size 4842
+oid sha256:0fd9d81ee48f1bc53f16eb05b90a6ae916b4a67afd89d33795522747c229b277
+size 5485