panyy
/
UnisKB
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix: Rearrange model dependency (#4324)

v3.2
shaohuzhang1 2025-11-06 14:24:48 +08:00 committed by GitHub
parent 3ab17c61ff
commit 1b464c4d8f
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
apps/models_provider/impl/local_model_provider/model/reranker/model.py
View File

@ -11,7 +11,6 @@ from typing import Sequence, Optional, Dict, Any
from langchain_core.callbacks import Callbacks from langchain_core.callbacks import Callbacks
from langchain_core.documents import Document, BaseDocumentCompressor from langchain_core.documents import Document, BaseDocumentCompressor
from transformers import AutoModelForSequenceClassification, AutoTokenizer
from models_provider.base_model_provider import MaxKBBaseModel 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): def __init__(self, model_name, cache_dir=None, **model_kwargs):
super().__init__() super().__init__()
from transformers import AutoModelForSequenceClassification, AutoTokenizer
self.model = model_name self.model = model_name
self.cache_dir = cache_dir self.cache_dir = cache_dir
self.model_kwargs = model_kwargs self.model_kwargs = model_kwargs