fix: llm_utils

This commit is contained in:
csunny 2023-05-31 14:15:36 +08:00
parent 09308bcdf0
commit a58eb5a95d

View File

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