From 1b464c4d8f9745331c46fe120982fa47d557607b Mon Sep 17 00:00:00 2001 From: shaohuzhang1 <80892890+shaohuzhang1@users.noreply.github.com> Date: Thu, 6 Nov 2025 14:24:48 +0800 Subject: [PATCH] fix: Rearrange model dependency (#4324) --- .../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 c5f1c67e0..a66776101 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 typing import Sequence, Optional, Dict, Any 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