marigold334 commited on
Commit
f74004c
1 Parent(s): 816e79a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -1
app.py CHANGED
@@ -27,7 +27,6 @@ class TTS:
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 == '감기걸린 은식':
 
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.decoder.skip()
31
  self.flowgenerator.eval()
32
  if model_variant == '감기걸린 은식':