Added Gemini to models init.py

pull/525/head
everettVT 1 year ago
parent dedf458a8a
commit 5062bf3e7a

@ -14,6 +14,7 @@ from swarms.models.nougat import Nougat # noqa: E402
from swarms.models.openai_embeddings import OpenAIEmbeddings from swarms.models.openai_embeddings import OpenAIEmbeddings
from swarms.models.openai_tts import OpenAITTS # noqa: E402 from swarms.models.openai_tts import OpenAITTS # noqa: E402
from swarms.models.palm import GooglePalm as Palm # noqa: E402 from swarms.models.palm import GooglePalm as Palm # noqa: E402
from swarms.models.gemini import Gemini
from swarms.models.popular_llms import Anthropic as Anthropic from swarms.models.popular_llms import Anthropic as Anthropic
from swarms.models.popular_llms import ( from swarms.models.popular_llms import (
AzureOpenAILLM as AzureOpenAI, AzureOpenAILLM as AzureOpenAI,
@ -46,6 +47,7 @@ __all__ = [
"BaseLLM", "BaseLLM",
"BaseMultiModalModel", "BaseMultiModalModel",
"Fuyu", "Fuyu",
"Gemini",
"GPT4VisionAPI", "GPT4VisionAPI",
"HuggingfaceLLM", "HuggingfaceLLM",
"Idefics", "Idefics",

Loading…
Cancel
Save