pengjunjie/ragflow
1
0
Fork 1
mirror of https://github.com/infiniflow/ragflow.git synced 2025-12-08 20:42:30 +08:00
Code Issues Packages Projects Releases Wiki Activity

add support for NVIDIA llm (#1645)

Browse Source 
### What problem does this PR solve?

add support for NVIDIA llm
### Type of change

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

---------

Co-authored-by: Zhedong Cen <cenzhedong2@126.com>
This commit is contained in:
黄腾
2024-07-23 10:43:09 +08:00
committed by GitHub
parent 95821f6fb6
commit b4a281eca1
8 changed files with 508 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
rag/llm/__init__.py
View File

@ -34,7 +34,8 @@ EmbeddingModel = {
"BAAI": DefaultEmbedding,
"Mistral": MistralEmbed,
"Bedrock": BedrockEmbed,
"Gemini":GeminiEmbed
"Gemini":GeminiEmbed,
"NVIDIA":NvidiaEmbed
}
@ -48,7 +49,8 @@ CvModel = {
"Moonshot": LocalCV,
'Gemini':GeminiCV,
'OpenRouter':OpenRouterCV,
"LocalAI":LocalAICV
"LocalAI":LocalAICV,
"NVIDIA":NvidiaCV
}
@ -71,7 +73,8 @@ ChatModel = {
"Bedrock": BedrockChat,
"Groq": GroqChat,
'OpenRouter':OpenRouterChat,
"StepFun":StepFunChat
"StepFun":StepFunChat,
"NVIDIA":NvidiaChat
}
@ -79,7 +82,8 @@ RerankModel = {
"BAAI": DefaultRerank,
"Jina": JinaRerank,
"Youdao": YoudaoRerank,
"Xinference": XInferenceRerank
"Xinference": XInferenceRerank,
"NVIDIA":NvidiaRerank
}