Spaces:

Hilley
/

ChatVC

Running

Hilley commited on Jun 2

Commit

55f0e01

•

1 Parent(s): 4511161

Update api.py

Files changed (1) hide show

api.py CHANGED Viewed

@@ -14,8 +14,8 @@ from models import SynthesizerTrn
 class OpenVoiceBaseClass(object):
     def __init__(self,
                 config_path,
-                #device='cuda:0'):
-                device="cpu"):
         #if 'cuda' in device:
         #    assert torch.cuda.is_available()

 class OpenVoiceBaseClass(object):
     def __init__(self,
                 config_path,
+                device='cuda:0'):
+                #device="cpu"):
         #if 'cuda' in device:
         #    assert torch.cuda.is_available()