Skip to content

Commit ba6d8c0

Browse files
committed
revert the main function to original and only preserve the inference endpoint
1 parent 2889c25 commit ba6d8c0

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

runtime/python/fastapi/server.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -88,14 +88,14 @@ async def inference_instruct2(tts_text: str = Form(), instruct_text: str = Form(
8888
default=50000)
8989
parser.add_argument('--model_dir',
9090
type=str,
91-
default='pretrained_models/CosyVoice2-0.5B',
91+
default='iic/CosyVoice-300M',
9292
help='local path or modelscope repo id')
9393
args = parser.parse_args()
9494
try:
9595
cosyvoice = CosyVoice(args.model_dir)
9696
except Exception:
9797
try:
98-
cosyvoice = CosyVoice2(args.model_dir, load_jit=True, load_trt=False)
98+
cosyvoice = CosyVoice2(args.model_dir)
9999
except Exception:
100100
raise TypeError('no valid model_type!')
101101
uvicorn.run(app, host="0.0.0.0", port=args.port)

0 commit comments

Comments
 (0)