diff --git a/embedchain/factory.py b/embedchain/factory.py index 9c0dea3e..fa46f3e9 100644 --- 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",