فهرست منبع

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

Abhishek Sharma 1 سال پیش
والد
کامیت
f2122ed696
1فایلهای تغییر یافته به همراه1 افزوده شده و 0 حذف شده
  1. 1 0
      embedchain/factory.py

+ 1 - 0
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",