github/DB-GPT
1
0
Fork 0
mirror of https://github.com/csunny/DB-GPT.git synced 2025-08-27 20:39:41 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat: Support baichuan-13B model

Browse Source
This commit is contained in:
FangYin Cheng 2023-07-25 00:53:28 +08:00
parent cf99fc67df
commit 7baaaf0849
2 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
pilot/model/adapter.py
View File

@ -12,6 +12,8 @@ from transformers import (
LlamaTokenizer, LlamaTokenizer,
BitsAndBytesConfig, BitsAndBytesConfig,
) )
from transformers.generation.utils import GenerationConfig
from pilot.configs.model_config import DEVICE from pilot.configs.model_config import DEVICE
from pilot.configs.config import Config from pilot.configs.config import Config

2
pilot/model/conversation.py
View File

@ -2,6 +2,8 @@
Fork from fastchat: https://github.com/lm-sys/FastChat/blob/main/fastchat/conversation.py Fork from fastchat: https://github.com/lm-sys/FastChat/blob/main/fastchat/conversation.py
Conversation prompt templates. Conversation prompt templates.
""" """
import dataclasses import dataclasses