Spaces:

FunAudioLLM
/

CosyVoice2-0.5B

Running on L4

R1ckShi commited on 9 days ago

Commit

b7a917f

•

1 Parent(s): 8bab248

update flow model to l20

Files changed (1) hide show

cosyvoice/cli/cosyvoice.py CHANGED Viewed

@@ -163,5 +163,5 @@ class CosyVoice2(CosyVoice):
         if load_onnx:
             self.model.load_onnx('{}/flow.decoder.estimator.fp32.onnx'.format(model_dir))
         if load_trt:
-            self.model.load_trt('{}/flow.decoder.estimator.fp16.A10.plan'.format(model_dir))
         del configs

         if load_onnx:
             self.model.load_onnx('{}/flow.decoder.estimator.fp32.onnx'.format(model_dir))
         if load_trt:
+            self.model.load_trt('{}/flow.decoder.estimator.fp16.l20.plan'.format(model_dir))
         del configs