diff --git a/test/models/test_fastchat_openai_llm.py b/test/models/test_fastchat_openai_llm.py index 1b09bf7..a0312be 100644 --- a/test/models/test_fastchat_openai_llm.py +++ b/test/models/test_fastchat_openai_llm.py @@ -32,7 +32,7 @@ async def dispatch(args: Namespace): if __name__ == '__main__': args = None - args = parser.parse_args(args=['--model-dir', '/media/checkpoint/', '--model', 'fastChatOpenAI', '--no-remote-model']) + args = parser.parse_args(args=['--model-dir', '/media/checkpoint/', '--model', 'fastchat-chatglm-6b', '--no-remote-model']) loop = asyncio.new_event_loop() asyncio.set_event_loop(loop)