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",