mirror of
https://github.com/csunny/DB-GPT.git
synced 2025-08-27 20:39:41 +00:00
feat: Support baichuan-13B model
This commit is contained in:
parent
cf99fc67df
commit
7baaaf0849
@ -12,6 +12,8 @@ from transformers import (
|
||||
LlamaTokenizer,
|
||||
BitsAndBytesConfig,
|
||||
)
|
||||
from transformers.generation.utils import GenerationConfig
|
||||
|
||||
from pilot.configs.model_config import DEVICE
|
||||
from pilot.configs.config import Config
|
||||
|
||||
|
@ -2,6 +2,8 @@
|
||||
Fork from fastchat: https://github.com/lm-sys/FastChat/blob/main/fastchat/conversation.py
|
||||
|
||||
Conversation prompt templates.
|
||||
|
||||
|
||||
"""
|
||||
|
||||
import dataclasses
|
||||
|
Loading…
Reference in New Issue
Block a user