github/DB-GPT
1
0
Fork 0
mirror of https://github.com/csunny/DB-GPT.git synced 2025-08-10 12:42:34 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: llm_utils

Browse Source
This commit is contained in:
csunny 2023-05-31 14:15:36 +08:00
parent 09308bcdf0
commit a58eb5a95d
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
pilot/model/guanaco_llm.py
View File

@ -4,7 +4,7 @@
import torch import torch
import transformers import transformers
from transformers import GenerationConfig 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): def guanaco_generate_output(model, tokenizer, params, device):