Feat: support huggingface re-rank model. (#5684)

### What problem does this PR solve?

#5658

### Type of change

- [x] New Feature (non-breaking change which adds functionality)
This commit is contained in:
Kevin Hu
2025-03-06 10:44:04 +08:00
committed by GitHub
parent 5f62f0c9d7
commit b8da2eeb69
3 changed files with 46 additions and 7 deletions

View File

@ -107,6 +107,7 @@ from .cv_model import (
YiCV,
HunyuanCV,
)
from .rerank_model import (
LocalAIRerank,
DefaultRerank,
@ -123,7 +124,9 @@ from .rerank_model import (
VoyageRerank,
QWenRerank,
GPUStackRerank,
HuggingfaceRerank,
)
from .sequence2txt_model import (
GPTSeq2txt,
QWenSeq2txt,
@ -132,6 +135,7 @@ from .sequence2txt_model import (
TencentCloudSeq2txt,
GPUStackSeq2txt,
)
from .tts_model import (
FishAudioTTS,
QwenTTS,
@ -255,6 +259,7 @@ RerankModel = {
"Voyage AI": VoyageRerank,
"Tongyi-Qianwen": QWenRerank,
"GPUStack": GPUStackRerank,
"HuggingFace": HuggingfaceRerank,
}
Seq2txtModel = {