scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-08 22:21:35 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 78

add vertex FT spec

Browse source
This commit is contained in:
Ishaan Jaff 2024-08-02 17:24:25 -07:00
parent 653cf55850
commit d364d76bd0
1 changed files with 23 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

23
litellm/types/llms/vertex_ai.py
View file

@ -260,3 +260,26 @@ class GenerateContentResponseBody(TypedDict, total=False):
candidates: Required[List[Candidates]]
promptFeedback: PromptFeedback
usageMetadata: Required[UsageMetadata]
class FineTunesupervisedTuningSpec(TypedDict, total=False):
training_dataset_uri: str
validation_dataset: Optional[str]
epoch_count: Optional[int]
learning_rate_multiplier: Optional[float]
tuned_model_display_name: Optional[str]
adapter_size: Optional[
Literal[
"ADAPTER_SIZE_UNSPECIFIED",
"ADAPTER_SIZE_ONE",
"ADAPTER_SIZE_FOUR",
"ADAPTER_SIZE_EIGHT",
"ADAPTER_SIZE_SIXTEEN",
]
]
class FineTuneJobCreate(TypedDict, total=False):
baseModel: str
supervisedTuningSpec: FineTunesupervisedTuningSpec
tunedModelDisplayName: str