From 9958c52f773397c0e53a8bec8a30f17864826218 Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Sat, 3 Aug 2024 08:56:54 -0700 Subject: [PATCH] fix vertex ai --- litellm/llms/fine_tuning_apis/vertex_ai.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/litellm/llms/fine_tuning_apis/vertex_ai.py b/litellm/llms/fine_tuning_apis/vertex_ai.py index 4c5938e68e9..f370652d269 100644 --- a/litellm/llms/fine_tuning_apis/vertex_ai.py +++ b/litellm/llms/fine_tuning_apis/vertex_ai.py @@ -88,7 +88,7 @@ class VertexFineTuningAPI(VertexLLM): def convert_openai_request_to_vertex( self, create_fine_tuning_job_data: FineTuningJobCreate, **kwargs - ) -> FineTuningJobCreate: + ) -> FineTuneJobCreate: """ convert request from OpenAI format to Vertex format https://cloud.google.com/vertex-ai/generative-ai/docs/model-reference/tuning @@ -122,7 +122,7 @@ class VertexFineTuningAPI(VertexLLM): self, fine_tuning_url: str, headers: dict, - request_data: FineTuningJobCreate, + request_data: FineTuneJobCreate, ): from litellm.fine_tuning.main import FineTuningJob