scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-12 23:01:41 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 182

fix vertex ai

Browse source
This commit is contained in:
Ishaan Jaff 2024-08-03 08:56:54 -07:00
parent 4ca0464395
commit 9958c52f77
1 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
litellm/llms/fine_tuning_apis/vertex_ai.py
View file

@ -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