ddd commited on
Commit
c99418d
1 Parent(s): 763ca1e

fix gradio on space

Browse files
Files changed (1) hide show
  1. inference/svs/ds_e2e.py +1 -1
inference/svs/ds_e2e.py CHANGED
@@ -24,7 +24,7 @@ class DiffSingerE2EInfer(BaseSVSInfer):
24
  load_ckpt(model, hparams['work_dir'], 'model')
25
 
26
  if hparams.get('pe_enable') is not None and hparams['pe_enable']:
27
- self.pe = PitchExtractor().cuda()
28
  utils.load_ckpt(self.pe, hparams['pe_ckpt'], 'model', strict=True)
29
  self.pe.eval()
30
  return model
 
24
  load_ckpt(model, hparams['work_dir'], 'model')
25
 
26
  if hparams.get('pe_enable') is not None and hparams['pe_enable']:
27
+ self.pe = PitchExtractor().to(self.device)
28
  utils.load_ckpt(self.pe, hparams['pe_ckpt'], 'model', strict=True)
29
  self.pe.eval()
30
  return model