Spaces:

ASLP-lab
/

OSUM-EChat

Running on Zero

xlgeng commited on Aug 21

Commit

1979f4b

1 Parent(s): daa8d34

开始部署

Files changed (1) hide show

tts/cosyvoice/cli/cosyvoice.py CHANGED Viewed

@@ -24,7 +24,7 @@ from tts.cosyvoice.utils.file_utils import logging
 from tts.cosyvoice.utils.class_utils import get_model_type
-class CosyVoice(torch.Module):
     def __init__(self, model_dir,gpu_id=-1, load_jit=False, load_trt=False, fp16=False):
         super(CosyVoice, self).__init__()

 from tts.cosyvoice.utils.class_utils import get_model_type
+class CosyVoice(torch.nn.Module):
     def __init__(self, model_dir,gpu_id=-1, load_jit=False, load_trt=False, fp16=False):
         super(CosyVoice, self).__init__()