kevinwang676 commited on
Commit
77b40c1
·
verified ·
1 Parent(s): 6593d8b

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
- device = 'cuda' if torch.cuda.is_available() else 'cpu'
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",