github/DB-GPT
1
0
Fork 0
mirror of https://github.com/csunny/DB-GPT.git synced 2025-09-02 09:37:03 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat: Support baichuan-7B model

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

2
pilot/model/adapter.py
View File

@@ -12,8 +12,6 @@ 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,8 +2,6 @@
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