Skip to content

Commit 1b464c4

Browse files
authored
fix: Rearrange model dependency (#4324)
1 parent 3ab17c6 commit 1b464c4

File tree

1 file changed

+1
-1
lines changed
  • apps/models_provider/impl/local_model_provider/model/reranker

1 file changed

+1
-1
lines changed

apps/models_provider/impl/local_model_provider/model/reranker/model.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -11,7 +11,6 @@
1111

1212
from langchain_core.callbacks import Callbacks
1313
from langchain_core.documents import Document, BaseDocumentCompressor
14-
from transformers import AutoModelForSequenceClassification, AutoTokenizer
1514

1615
from models_provider.base_model_provider import MaxKBBaseModel
1716

@@ -25,6 +24,7 @@ class LocalReranker(MaxKBBaseModel, BaseDocumentCompressor):
2524

2625
def __init__(self, model_name, cache_dir=None, **model_kwargs):
2726
super().__init__()
27+
from transformers import AutoModelForSequenceClassification, AutoTokenizer
2828
self.model = model_name
2929
self.cache_dir = cache_dir
3030
self.model_kwargs = model_kwargs

0 commit comments

Comments
 (0)