diff --git a/runtime/python/fastapi/server.py b/runtime/python/fastapi/server.py index 6805714..08542df 100644 --- a/runtime/python/fastapi/server.py +++ b/runtime/python/fastapi/server.py @@ -88,14 +88,14 @@ if __name__ == '__main__': default=50000) parser.add_argument('--model_dir', type=str, - default='pretrained_models/CosyVoice2-0.5B', + default='iic/CosyVoice-300M', help='local path or modelscope repo id') args = parser.parse_args() try: cosyvoice = CosyVoice(args.model_dir) except Exception: try: - cosyvoice = CosyVoice2(args.model_dir, load_jit=True, load_trt=False) + cosyvoice = CosyVoice2(args.model_dir) except Exception: raise TypeError('no valid model_type!') uvicorn.run(app, host="0.0.0.0", port=args.port) \ No newline at end of file