kevinwang676 commited on
Commit
64258a3
1 Parent(s): cd9cb27

Update api.py

Browse files
Files changed (1) hide show
  1. api.py +2 -1
api.py CHANGED
@@ -14,7 +14,8 @@ from models import SynthesizerTrn
14
  class OpenVoiceBaseClass(object):
15
  def __init__(self,
16
  config_path,
17
- device='cuda:0'):
 
18
  if 'cuda' in device:
19
  assert torch.cuda.is_available()
20
 
 
14
  class OpenVoiceBaseClass(object):
15
  def __init__(self,
16
  config_path,
17
+ #device='cuda:0'):
18
+ devive="cpu"):
19
  if 'cuda' in device:
20
  assert torch.cuda.is_available()
21