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

2024-03-14 12:31:30 +05:30
parent 3616eaadb4
commit f2122ed696
1 changed files with 1 additions and 0 deletions
--- a/embedchain/factory.py
+++ b/embedchain/factory.py
@@ -25,6 +25,7 @@ class LlmFactory:
        "mistralai": "embedchain.llm.mistralai.MistralAILlm",
        "groq": "embedchain.llm.groq.GroqLlm",
        "nvidia": "embedchain.llm.nvidia.NvidiaLlm",
+        "vllm": "embedchain.llm.vllm.VLLM",
    }
    provider_to_config_class = {
        "embedchain": "embedchain.config.llm.base.BaseLlmConfig",