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