fix dir structure for tts

This commit is contained in:
Ishaan Jaff 2024-08-30 11:44:23 -07:00
parent 4466ec658f
commit c40e564779
2 changed files with 3 additions and 1 deletions

View file

@ -120,7 +120,6 @@ from .llms.prompt_templates.factory import (
) )
from .llms.sagemaker.sagemaker import SagemakerLLM from .llms.sagemaker.sagemaker import SagemakerLLM
from .llms.text_completion_codestral import CodestralTextCompletion from .llms.text_completion_codestral import CodestralTextCompletion
from .llms.text_to_speech.vertex_ai import VertexTextToSpeechAPI
from .llms.triton import TritonChatCompletion from .llms.triton import TritonChatCompletion
from .llms.vertex_ai_and_google_ai_studio import ( from .llms.vertex_ai_and_google_ai_studio import (
vertex_ai_anthropic, vertex_ai_anthropic,
@ -135,6 +134,9 @@ from .llms.vertex_ai_and_google_ai_studio.gemini.vertex_and_google_ai_studio_gem
from .llms.vertex_ai_and_google_ai_studio.multimodal_embeddings.embedding_handler import ( from .llms.vertex_ai_and_google_ai_studio.multimodal_embeddings.embedding_handler import (
VertexMultimodalEmbedding, VertexMultimodalEmbedding,
) )
from .llms.vertex_ai_and_google_ai_studio.text_to_speech.text_to_speech_handler import (
VertexTextToSpeechAPI,
)
from .llms.vertex_ai_and_google_ai_studio.vertex_ai_partner_models.main import ( from .llms.vertex_ai_and_google_ai_studio.vertex_ai_partner_models.main import (
VertexAIPartnerModels, VertexAIPartnerModels,
) )