Mahiruoshi commited on
Commit
3ad53be
1 Parent(s): 94c2d64

Update inference.py

Browse files
Files changed (1) hide show
  1. inference.py +3 -7
inference.py CHANGED
@@ -120,12 +120,8 @@ idols = ["高咲侑","歩夢","かすみ","しずく","果林","愛","彼方","
120
 
121
 
122
 
123
- Device = input("设置运行时类型")
124
- if Device == "cpu":
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("C:/Users/24293/机器学习-/MB-iSTFT-VITS/logs/nijigaku/G_1330000.pth", net_g_ms, None)
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