Spaces:
Runtime error
Runtime error
Update app.py
#24
by
marigold334
- opened
app.py
CHANGED
@@ -26,7 +26,8 @@ class TTS:
|
|
26 |
elif model_variant == '태연':
|
27 |
last_chpt1 = './log/Taeyeon/Glow_TTS_400000.pt'
|
28 |
check_point = torch.load(last_chpt1, map_location = device)
|
29 |
-
self.flowgenerator.load_state_dict(check_point['generator'])
|
|
|
30 |
self.flowgenerator.decoder.skip()
|
31 |
self.flowgenerator.eval()
|
32 |
if model_variant == '감기걸린 은식':
|
|
|
26 |
elif model_variant == '태연':
|
27 |
last_chpt1 = './log/Taeyeon/Glow_TTS_400000.pt'
|
28 |
check_point = torch.load(last_chpt1, map_location = device)
|
29 |
+
self.flowgenerator.load_state_dict(check_point['generator' if model_variant != '태연' else 'model'])
|
30 |
+
self.flowgenerator.load_state_dict(check_point['generator'])
|
31 |
self.flowgenerator.decoder.skip()
|
32 |
self.flowgenerator.eval()
|
33 |
if model_variant == '감기걸린 은식':
|