feat: Support baichuan-7B model

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

View File

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

View File

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