You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
99 lines
2.8 KiB
99 lines
2.8 KiB
############################################ LLMs
|
|
from swarms.models.base_llm import AbstractLLM # noqa: E402
|
|
from swarms.models.anthropic import Anthropic # noqa: E402
|
|
from swarms.models.petals import Petals # noqa: E402
|
|
from swarms.models.mistral import Mistral # noqa: E402
|
|
from swarms.models.openai_models import (
|
|
OpenAI,
|
|
AzureOpenAI,
|
|
OpenAIChat,
|
|
) # 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 (
|
|
WizardLLMStoryTeller,
|
|
) # noqa: E402
|
|
from swarms.models.mpt import MPT7B # noqa: E402
|
|
from swarms.models.mixtral import Mixtral # noqa: E402
|
|
# from swarms.models.modelscope_pipeline import ModelScopePipeline
|
|
# from swarms.models.modelscope_llm import (
|
|
# ModelScopeAutoModel,
|
|
# ) # noqa: E402
|
|
from swarms.models.together import TogetherLLM # noqa: E402
|
|
|
|
################# MultiModal Models
|
|
from swarms.models.base_multimodal_model import (
|
|
BaseMultiModalModel,
|
|
) # noqa: E402
|
|
from swarms.models.idefics import Idefics # noqa: E402
|
|
from swarms.models.vilt import Vilt # noqa: E402
|
|
from swarms.models.nougat import Nougat # noqa: E402
|
|
from swarms.models.layoutlm_document_qa import (
|
|
LayoutLMDocumentQA,
|
|
) # noqa: E402
|
|
from swarms.models.gpt4_vision_api import GPT4VisionAPI # noqa: E402
|
|
from swarms.models.openai_tts import OpenAITTS # noqa: E402
|
|
from swarms.models.gemini import Gemini # noqa: E402
|
|
from swarms.models.gigabind import Gigabind # noqa: E402
|
|
from swarms.models.zeroscope import ZeroscopeTTV # noqa: E402
|
|
|
|
|
|
# from swarms.models.dalle3 import Dalle3
|
|
# from swarms.models.distilled_whisperx import DistilWhisperModel # noqa: E402
|
|
# from swarms.models.whisperx_model import WhisperX # noqa: E402
|
|
# from swarms.models.kosmos_two import Kosmos # noqa: E402
|
|
# from swarms.models.cog_agent import CogAgent # noqa: E402
|
|
|
|
|
|
############## Types
|
|
from swarms.models.types import (
|
|
TextModality,
|
|
ImageModality,
|
|
AudioModality,
|
|
VideoModality,
|
|
MultimodalData,
|
|
) # noqa: E402
|
|
|
|
__all__ = [
|
|
"AbstractLLM",
|
|
"Anthropic",
|
|
"Petals",
|
|
"Mistral",
|
|
"OpenAI",
|
|
"AzureOpenAI",
|
|
"OpenAIChat",
|
|
"Zephyr",
|
|
"BaseMultiModalModel",
|
|
"Idefics",
|
|
# "Kosmos",
|
|
"Vilt",
|
|
"Nougat",
|
|
"LayoutLMDocumentQA",
|
|
"BioGPT",
|
|
"HuggingfaceLLM",
|
|
"MPT7B",
|
|
"WizardLLMStoryTeller",
|
|
# "GPT4Vision",
|
|
# "Dalle3",
|
|
# "DistilWhisperModel",
|
|
"GPT4VisionAPI",
|
|
# "vLLM",
|
|
"OpenAITTS",
|
|
"Gemini",
|
|
"Gigabind",
|
|
"Mixtral",
|
|
"ZeroscopeTTV",
|
|
"TextModality",
|
|
"ImageModality",
|
|
"AudioModality",
|
|
"VideoModality",
|
|
"MultimodalData",
|
|
# "CogAgent",
|
|
# "ModelScopePipeline",
|
|
# "ModelScopeAutoModel",
|
|
"TogetherLLM",
|
|
]
|