Simonlob commited on
Commit
8efe532
·
verified ·
1 Parent(s): 231e1c2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -90,7 +90,7 @@ def get_inference(text, n_timesteps=20, mel_temp = 0.667, length_scale=0.8, spk=
90
  return synthesise_mel(text, text_lengths, n_timesteps, mel_temp, length_scale, spk)
91
 
92
 
93
- device = torch.device("cpu") #torch.device("cuda")
94
  model_path = './checkpoints/checkpoint.ckpt'
95
  vocoder_path = './checkpoints/generator'
96
  model = load_matcha(model_path, device)
 
90
  return synthesise_mel(text, text_lengths, n_timesteps, mel_temp, length_scale, spk)
91
 
92
 
93
+ device = torch.device("cuda")
94
  model_path = './checkpoints/checkpoint.ckpt'
95
  vocoder_path = './checkpoints/generator'
96
  model = load_matcha(model_path, device)