kemuriririn commited on
Commit
f58a4a7
·
1 Parent(s): 3c02e07

add gpu tags

Browse files
Files changed (1) hide show
  1. cosyvoice/cli/cosyvoice.py +1 -1
cosyvoice/cli/cosyvoice.py CHANGED
@@ -23,7 +23,7 @@ from cosyvoice.utils.file_utils import logging
23
  import spaces
24
 
25
  class CosyVoice:
26
-
27
  def __init__(self, model_dir, load_jit=True, load_onnx=False, fp16=True):
28
  instruct = True if '-Instruct' in model_dir else False
29
  self.model_dir = model_dir
 
23
  import spaces
24
 
25
  class CosyVoice:
26
+ @spaces.GPU
27
  def __init__(self, model_dir, load_jit=True, load_onnx=False, fp16=True):
28
  instruct = True if '-Instruct' in model_dir else False
29
  self.model_dir = model_dir