add huggingface model (#2624)

### What problem does this PR solve?

#2469

### Type of change

- [x] New Feature (non-breaking change which adds functionality)

---------

Co-authored-by: Kevin Hu <kevinhu.sh@gmail.com>
This commit is contained in:
JobSmithManipulation
2024-09-27 19:15:38 +08:00
committed by GitHub
parent 1b2f66fc11
commit 96f56a3c43
9 changed files with 120 additions and 13 deletions

View File

@ -18,7 +18,7 @@ from .chat_model import *
from .cv_model import *
from .rerank_model import *
from .sequence2txt_model import *
from .tts_model import *
from .tts_model import *
EmbeddingModel = {
"Ollama": OllamaEmbed,
@ -46,7 +46,8 @@ EmbeddingModel = {
"SILICONFLOW": SILICONFLOWEmbed,
"Replicate": ReplicateEmbed,
"BaiduYiyan": BaiduYiyanEmbed,
"Voyage AI": VoyageEmbed
"Voyage AI": VoyageEmbed,
"HuggingFace":HuggingFaceEmbed,
}