Spaces:
Runtime error
Runtime error
ddd
commited on
Commit
•
c99418d
1
Parent(s):
763ca1e
fix gradio on space
Browse files- 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().
|
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
|