marigold334 commited on
Commit
5702878
1 Parent(s): c5fff6b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
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 == '감기걸린 은식':