Mahiruoshi
commited on
Commit
•
3ad53be
1
Parent(s):
94c2d64
Update inference.py
Browse files- inference.py +3 -7
inference.py
CHANGED
@@ -120,12 +120,8 @@ idols = ["高咲侑","歩夢","かすみ","しずく","果林","愛","彼方","
|
|
120 |
|
121 |
|
122 |
|
123 |
-
|
124 |
-
|
125 |
-
dev = torch.device("cpu")
|
126 |
-
else:
|
127 |
-
dev = torch.device("cuda:0")
|
128 |
-
hps_ms = utils.get_hparams_from_file("C:/Users/24293/机器学习-/MB-iSTFT-VITS/logs/nijigaku/config.json")
|
129 |
net_g_ms = SynthesizerTrn(
|
130 |
len(symbols),
|
131 |
hps_ms.data.filter_length // 2 + 1,
|
@@ -134,7 +130,7 @@ net_g_ms = SynthesizerTrn(
|
|
134 |
**hps_ms.model).to(dev)
|
135 |
_ = net_g_ms.eval()
|
136 |
|
137 |
-
_ = utils.load_checkpoint("
|
138 |
|
139 |
app = gr.Blocks()
|
140 |
|
|
|
120 |
|
121 |
|
122 |
|
123 |
+
dev = torch.device("cpu")
|
124 |
+
hps_ms = utils.get_hparams_from_file("config.json")
|
|
|
|
|
|
|
|
|
125 |
net_g_ms = SynthesizerTrn(
|
126 |
len(symbols),
|
127 |
hps_ms.data.filter_length // 2 + 1,
|
|
|
130 |
**hps_ms.model).to(dev)
|
131 |
_ = net_g_ms.eval()
|
132 |
|
133 |
+
_ = utils.load_checkpoint("G_1415000.pth", net_g_ms, None)
|
134 |
|
135 |
app = gr.Blocks()
|
136 |
|