klas/t6_mem0
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

[Fix] Added missing provider for 'vllm' (#1316)

Browse Source
This commit is contained in:
Abhishek Sharma
2024-03-14 12:31:30 +05:30
committed by GitHub
parent 3616eaadb4
commit f2122ed696
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
embedchain/factory.py
View File

@@ -25,6 +25,7 @@ class LlmFactory:
"mistralai": "embedchain.llm.mistralai.MistralAILlm", "mistralai": "embedchain.llm.mistralai.MistralAILlm",
"groq": "embedchain.llm.groq.GroqLlm", "groq": "embedchain.llm.groq.GroqLlm",
"nvidia": "embedchain.llm.nvidia.NvidiaLlm", "nvidia": "embedchain.llm.nvidia.NvidiaLlm",
"vllm": "embedchain.llm.vllm.VLLM",
} }
provider_to_config_class = { provider_to_config_class = {
"embedchain": "embedchain.config.llm.base.BaseLlmConfig", "embedchain": "embedchain.config.llm.base.BaseLlmConfig",