Spaces:
Paused
Paused
Update inference/m4singer/base_svs_infer.py
Browse files
inference/m4singer/base_svs_infer.py
CHANGED
@@ -18,10 +18,10 @@ import spaces
|
|
18 |
|
19 |
class BaseSVSInfer:
|
20 |
def __init__(self, hparams, device=None):
|
21 |
-
if device is None:
|
22 |
-
|
23 |
self.hparams = hparams
|
24 |
-
self.device = device
|
25 |
|
26 |
phone_list = ["<AP>", "<SP>", "a", "ai", "an", "ang", "ao", "b", "c", "ch", "d", "e", "ei", "en", "eng", "er", "f", "g", "h",
|
27 |
"i", "ia", "ian", "iang", "iao", "ie", "in", "ing", "iong", "iou", "j", "k", "l", "m", "n", "o", "ong", "ou",
|
|
|
18 |
|
19 |
class BaseSVSInfer:
|
20 |
def __init__(self, hparams, device=None):
|
21 |
+
#if device is None:
|
22 |
+
# device = 'cuda' if torch.cuda.is_available() else 'cpu'
|
23 |
self.hparams = hparams
|
24 |
+
self.device = "cuda" #device
|
25 |
|
26 |
phone_list = ["<AP>", "<SP>", "a", "ai", "an", "ang", "ao", "b", "c", "ch", "d", "e", "ei", "en", "eng", "er", "f", "g", "h",
|
27 |
"i", "ia", "ian", "iang", "iao", "ie", "in", "ing", "iong", "iou", "j", "k", "l", "m", "n", "o", "ong", "ou",
|