From 4b1c0a36f8cddd1a0df138466ce4ce29608363e8 Mon Sep 17 00:00:00 2001 From: FangYin Cheng Date: Tue, 25 Jul 2023 18:39:07 +0800 Subject: [PATCH] feat: Support baichuan-7B model --- pilot/model/adapter.py | 2 -- pilot/model/conversation.py | 2 -- 2 files changed, 4 deletions(-) diff --git a/pilot/model/adapter.py b/pilot/model/adapter.py index 05643c734..900d51d4a 100644 --- a/pilot/model/adapter.py +++ b/pilot/model/adapter.py @@ -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 diff --git a/pilot/model/conversation.py b/pilot/model/conversation.py index d29e73da3..11ca03ed8 100644 --- a/pilot/model/conversation.py +++ b/pilot/model/conversation.py @@ -2,8 +2,6 @@ Fork from fastchat: https://github.com/lm-sys/FastChat/blob/main/fastchat/conversation.py Conversation prompt templates. - - """ import dataclasses