We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 3ab17c6 commit 1b464c4Copy full SHA for 1b464c4
apps/models_provider/impl/local_model_provider/model/reranker/model.py
@@ -11,7 +11,6 @@
11
12
from langchain_core.callbacks import Callbacks
13
from langchain_core.documents import Document, BaseDocumentCompressor
14
-from transformers import AutoModelForSequenceClassification, AutoTokenizer
15
16
from models_provider.base_model_provider import MaxKBBaseModel
17
@@ -25,6 +24,7 @@ class LocalReranker(MaxKBBaseModel, BaseDocumentCompressor):
25
24
26
def __init__(self, model_name, cache_dir=None, **model_kwargs):
27
super().__init__()
+ from transformers import AutoModelForSequenceClassification, AutoTokenizer
28
self.model = model_name
29
self.cache_dir = cache_dir
30
self.model_kwargs = model_kwargs
0 commit comments