Spaces:
Running
Running
marigold334
commited on
Commit
β’
ddbc9f0
1
Parent(s):
ecc0d3c
Update app.py (#29)
Browse files- Update app.py (64e6fa32be4852c0b9a923d653b45b33c4dec69c)
app.py
CHANGED
@@ -25,7 +25,7 @@ class TTS:
|
|
25 |
elif model_variant == 'KSS':
|
26 |
last_chpt1 = './log/KSS/Glow_TTS_00280641.pt'
|
27 |
elif model_variant == 'νμ°':
|
28 |
-
last_chpt1 = './log/Taeyeon/
|
29 |
check_point = torch.load(last_chpt1, map_location = device)
|
30 |
self.flowgenerator.load_state_dict(check_point['generator' if model_variant != 'νμ°' else 'model'])
|
31 |
self.flowgenerator.decoder.skip()
|
@@ -35,7 +35,7 @@ class TTS:
|
|
35 |
elif model_variant == 'KSS':
|
36 |
last_chpt2 = './log/KSS/HiFi_GAN_00135000.pt'
|
37 |
elif model_variant == 'νμ°':
|
38 |
-
last_chpt2 = './log/Taeyeon/
|
39 |
check_point = torch.load(last_chpt2, map_location = device)
|
40 |
self.voicegenerator.load_state_dict(check_point['gen_model'])
|
41 |
self.voicegenerator.eval()
|
|
|
25 |
elif model_variant == 'KSS':
|
26 |
last_chpt1 = './log/KSS/Glow_TTS_00280641.pt'
|
27 |
elif model_variant == 'νμ°':
|
28 |
+
last_chpt1 = './log/Taeyeon/Glow_TTS_337000.pt'
|
29 |
check_point = torch.load(last_chpt1, map_location = device)
|
30 |
self.flowgenerator.load_state_dict(check_point['generator' if model_variant != 'νμ°' else 'model'])
|
31 |
self.flowgenerator.decoder.skip()
|
|
|
35 |
elif model_variant == 'KSS':
|
36 |
last_chpt2 = './log/KSS/HiFi_GAN_00135000.pt'
|
37 |
elif model_variant == 'νμ°':
|
38 |
+
last_chpt2 = './log/Taeyeon/HiFi_GAN_400000.pt'
|
39 |
check_point = torch.load(last_chpt2, map_location = device)
|
40 |
self.voicegenerator.load_state_dict(check_point['gen_model'])
|
41 |
self.voicegenerator.eval()
|