mirror of
https://github.com/csunny/DB-GPT.git
synced 2025-08-09 04:08:10 +00:00
fix: llm_utils
This commit is contained in:
parent
09308bcdf0
commit
a58eb5a95d
@ -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):
|
||||
|
Loading…
Reference in New Issue
Block a user