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):