From 38438c2266c141b2b8868e6f3be987c515e415ed Mon Sep 17 00:00:00 2001 From: Kye Date: Sun, 31 Dec 2023 01:00:21 -0500 Subject: [PATCH] [FEAT]][Vllm] --- pyproject.toml | 2 +- swarms/models/__init__.py | 6 +++--- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/pyproject.toml b/pyproject.toml index e5fb11f0..6c0d109a 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -4,7 +4,7 @@ build-backend = "poetry.core.masonry.api" [tool.poetry] name = "swarms" -version = "3.1.1" +version = "3.1.2" description = "Swarms - Pytorch" license = "MIT" authors = ["Kye Gomez "] diff --git a/swarms/models/__init__.py b/swarms/models/__init__.py index cc02b4f4..9ab84e5e 100644 --- a/swarms/models/__init__.py +++ b/swarms/models/__init__.py @@ -9,8 +9,8 @@ from swarms.models.openai_models import ( OpenAIChat, ) # noqa: E402 -# from swarms.models.vllm import vLLM # noqa: E402 -# from swarms.models.zephyr import Zephyr # noqa: E402 +from swarms.models.vllm import vLLM # noqa: E402 +from swarms.models.zephyr import Zephyr # noqa: E402 from swarms.models.biogpt import BioGPT # noqa: E402 from swarms.models.huggingface import HuggingfaceLLM # noqa: E402 from swarms.models.wizard_storytelling import ( @@ -64,7 +64,7 @@ __all__ = [ # "Dalle3", # "DistilWhisperModel", "GPT4VisionAPI", - # "vLLM", + "vLLM", "OpenAITTS", "Gemini", "Gigabind",