From e1c2ff41a04856fb97f2defaa8a1c4556ff214fb Mon Sep 17 00:00:00 2001
From: BUAADreamer <1428195643@qq.com>
Date: Wed, 15 May 2024 12:48:18 +0800
Subject: [PATCH] rm extra import

Former-commit-id: db1622f76b0fe9d669af206299ecec10954647af
---
 src/llmtuner/model/utils/visual.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/llmtuner/model/utils/visual.py b/src/llmtuner/model/utils/visual.py
index b8696096..1f770861 100644
--- a/src/llmtuner/model/utils/visual.py
+++ b/src/llmtuner/model/utils/visual.py
@@ -8,7 +8,7 @@ from ...extras.logging import get_logger
 
 
 if TYPE_CHECKING:
-    from transformers import LlavaConfig, PretrainedConfig, PreTrainedModel, LlavaForConditionalGeneration
+    from transformers import LlavaConfig, PretrainedConfig, PreTrainedModel
 
     from ...hparams import ModelArguments