mirror of
https://github.com/csunny/DB-GPT.git
synced 2025-09-12 12:37:14 +00:00
fix: llm_utils
This commit is contained in:
@@ -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):
|
||||
|
Reference in New Issue
Block a user