From 47b55b1c91f0b7e93ef7ef4aeb58fad03b193ad3 Mon Sep 17 00:00:00 2001 From: csunny Date: Fri, 12 May 2023 20:47:46 +0800 Subject: [PATCH] fix --- pilot/model/llm_utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pilot/model/llm_utils.py b/pilot/model/llm_utils.py index ee4e3b6e9..196246118 100644 --- a/pilot/model/llm_utils.py +++ b/pilot/model/llm_utils.py @@ -4,7 +4,7 @@ from typing import List, Optional from pilot.model.base import Message from pilot.configs.config import Config -from pilot.server.vicuna_server import generate_output +from pilot.server.llmserver import generate_output def create_chat_completion( messages: List[Message], # type: ignore