Spaces:

kevinwang676
/

M4Singer-Zero

Paused

kevinwang676 commited on Jun 26, 2024

Commit

fa3712e

verified ·

1 Parent(s): e3d69cb

Update inference/m4singer/base_svs_infer.py

Files changed (1) hide show

inference/m4singer/base_svs_infer.py CHANGED Viewed

@@ -17,6 +17,7 @@ import re
 import spaces
 class BaseSVSInfer:
     def __init__(self, hparams, device=None):
         if device is None:
             device = 'cuda' if torch.cuda.is_available() else 'cpu'

 import spaces
 class BaseSVSInfer:
+    @spaces.GPU
     def __init__(self, hparams, device=None):
         if device is None:
             device = 'cuda' if torch.cuda.is_available() else 'cpu'