From ba6d8c07ba0ba3e0c459afa2e51e016f34712d7b Mon Sep 17 00:00:00 2001 From: c4fun Date: Fri, 7 Feb 2025 16:28:30 +0800 Subject: [PATCH] revert the main function to original and only preserve the inference endpoint --- runtime/python/fastapi/server.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) 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