From 1552c105d9e335947aa783d6471559d96d31c7f8 Mon Sep 17 00:00:00 2001 From: shaohuzhang1 Date: Thu, 6 Nov 2025 14:23:58 +0800 Subject: [PATCH] fix: Rearrange model dependency --- .../impl/local_model_provider/model/reranker/model.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/apps/models_provider/impl/local_model_provider/model/reranker/model.py b/apps/models_provider/impl/local_model_provider/model/reranker/model.py index c5f1c67e085..a66776101b4 100644 --- a/apps/models_provider/impl/local_model_provider/model/reranker/model.py +++ b/apps/models_provider/impl/local_model_provider/model/reranker/model.py @@ -11,7 +11,6 @@ from langchain_core.callbacks import Callbacks from langchain_core.documents import Document, BaseDocumentCompressor -from transformers import AutoModelForSequenceClassification, AutoTokenizer from models_provider.base_model_provider import MaxKBBaseModel @@ -25,6 +24,7 @@ class LocalReranker(MaxKBBaseModel, BaseDocumentCompressor): def __init__(self, model_name, cache_dir=None, **model_kwargs): super().__init__() + from transformers import AutoModelForSequenceClassification, AutoTokenizer self.model = model_name self.cache_dir = cache_dir self.model_kwargs = model_kwargs