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.
61 lines
1.8 KiB
61 lines
1.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.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
|
|
|
|
# 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.gpt4v import GPT4Vision
|
|
# 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
|
|
|
|
__all__ = [
|
|
"AbstractLLM",
|
|
"Anthropic",
|
|
"Petals",
|
|
"Mistral",
|
|
"OpenAI",
|
|
"AzureOpenAI",
|
|
"OpenAIChat",
|
|
"Zephyr",
|
|
"BaseMultiModalModel",
|
|
"Idefics",
|
|
# "Kosmos",
|
|
"Vilt",
|
|
"Nougat",
|
|
"LayoutLMDocumentQA",
|
|
"BioGPT",
|
|
"HuggingfaceLLM",
|
|
"MPT7B",
|
|
"WizardLLMStoryTeller",
|
|
# "GPT4Vision",
|
|
# "Dalle3",
|
|
# "DistilWhisperModel",
|
|
"GPT4VisionAPI",
|
|
]
|