marigold334 commited on
Commit
5f9814f
1 Parent(s): b1ee5a2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -17,7 +17,7 @@ class TTS:
17
  global device
18
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
19
  torch.cuda.manual_seed(1234) if torch.cuda.is_available() else None
20
- self.flowgenerator = Glow_model(n_vocab = 70 , h_c= 192, f_c = 768, f_c_dp = 256, out_c = 80, k_s = 3, k_s_dec = 5, heads=2, layers_enc = 6).to(device) if model_variant != '태연' else T_Glow_model().to(device)
21
  self.voicegenerator = GAN_model().to(device)
22
  if model_variant == '감기걸린 은식':
23
  last_chpt1 = './log/1038_eunsik_01/Glow_TTS_00289602.pt'
 
17
  global device
18
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
19
  torch.cuda.manual_seed(1234) if torch.cuda.is_available() else None
20
+ self.flowgenerator = Glow_model(n_vocab = 70 , h_c= 192, f_c = 768, f_c_dp = 256, out_c = 80, k_s = 3, k_s_dec = 5, heads=2, layers_enc = 6).to(device) if model_variant != '술취한 태연' else T_Glow_model().to(device)
21
  self.voicegenerator = GAN_model().to(device)
22
  if model_variant == '감기걸린 은식':
23
  last_chpt1 = './log/1038_eunsik_01/Glow_TTS_00289602.pt'