pull/362/head
Kye 12 months ago
parent 9ef3e68e27
commit 73c946d4ba

@ -4,7 +4,7 @@ build-backend = "poetry.core.masonry.api"
[tool.poetry] [tool.poetry]
name = "swarms" name = "swarms"
version = "3.6.5" version = "3.6.6"
description = "Swarms - Pytorch" description = "Swarms - Pytorch"
license = "MIT" license = "MIT"
authors = ["Kye Gomez <kye@apac.ai>"] authors = ["Kye Gomez <kye@apac.ai>"]
@ -71,7 +71,6 @@ pgvector = "*"
qdrant-client = "*" qdrant-client = "*"
sentence-transformers = "*" sentence-transformers = "*"
peft = "*" peft = "*"
modelscope = "1.10.0"
psutil = "*" psutil = "*"

@ -59,5 +59,4 @@ mkdocs-material
mkdocs-glightbox mkdocs-glightbox
pre-commit==3.2.2 pre-commit==3.2.2
peft peft
modelscope
psutil psutil

@ -18,10 +18,10 @@ from swarms.models.wizard_storytelling import (
) # noqa: E402 ) # noqa: E402
from swarms.models.mpt import MPT7B # noqa: E402 from swarms.models.mpt import MPT7B # noqa: E402
from swarms.models.mixtral import Mixtral # noqa: E402 from swarms.models.mixtral import Mixtral # noqa: E402
from swarms.models.modelscope_pipeline import ModelScopePipeline # from swarms.models.modelscope_pipeline import ModelScopePipeline
from swarms.models.modelscope_llm import ( # from swarms.models.modelscope_llm import (
ModelScopeAutoModel, # ModelScopeAutoModel,
) # noqa: E402 # ) # noqa: E402
from swarms.models.together import TogetherLLM # noqa: E402 from swarms.models.together import TogetherLLM # noqa: E402
################# MultiModal Models ################# MultiModal Models
@ -45,7 +45,7 @@ from swarms.models.zeroscope import ZeroscopeTTV # noqa: E402
# from swarms.models.distilled_whisperx import DistilWhisperModel # noqa: E402 # from swarms.models.distilled_whisperx import DistilWhisperModel # noqa: E402
# from swarms.models.whisperx_model import WhisperX # noqa: E402 # from swarms.models.whisperx_model import WhisperX # noqa: E402
# from swarms.models.kosmos_two import Kosmos # noqa: E402 # from swarms.models.kosmos_two import Kosmos # noqa: E402
from swarms.models.cog_agent import CogAgent # noqa: E402 # from swarms.models.cog_agent import CogAgent # noqa: E402
############## Types ############## Types
@ -91,8 +91,8 @@ __all__ = [
"AudioModality", "AudioModality",
"VideoModality", "VideoModality",
"MultimodalData", "MultimodalData",
"CogAgent", # "CogAgent",
"ModelScopePipeline", # "ModelScopePipeline",
"ModelScopeAutoModel", # "ModelScopeAutoModel",
"TogetherLLM", "TogetherLLM",
] ]

Loading…
Cancel
Save