From a58eb5a95d810abc1ff35c3cfed2a47d75b2fecb Mon Sep 17 00:00:00 2001 From: csunny Date: Wed, 31 May 2023 14:15:36 +0800 Subject: [PATCH] fix: llm_utils --- pilot/model/guanaco_llm.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pilot/model/guanaco_llm.py b/pilot/model/guanaco_llm.py index 84fd3eede..df4f86dce 100644 --- a/pilot/model/guanaco_llm.py +++ b/pilot/model/guanaco_llm.py @@ -4,7 +4,7 @@ import torch import transformers from transformers import GenerationConfig -from llm_utils import Iteratorize, Stream +from pilot.model.llm_utils import Iteratorize, Stream def guanaco_generate_output(model, tokenizer, params, device):