forked from phoenix/litellm-mirror
Merge pull request #5030 from BerriAI/litellm_add_vertex_ft_proxy
[Feat] Add support for Vertex AI Fine tuning on LiteLLM Proxy
This commit is contained in:
commit
9dffe23108
2 changed files with 5 additions and 3 deletions
|
@ -34,6 +34,10 @@ finetune_settings:
|
|||
api_version: "2023-03-15-preview"
|
||||
- custom_llm_provider: openai
|
||||
api_key: os.environ/OPENAI_API_KEY
|
||||
- custom_llm_provider: "vertex_ai"
|
||||
vertex_project: "adroit-crow-413218"
|
||||
vertex_location: "us-central1"
|
||||
vertex_credentials: "/Users/ishaanjaffer/Downloads/adroit-crow-413218-a956eef1a2a8.json"
|
||||
|
||||
# for /files endpoints
|
||||
files_settings:
|
||||
|
@ -54,5 +58,3 @@ general_settings:
|
|||
# google cloud run maximum repsonses size is 32MB
|
||||
max_response_size_mb: 10
|
||||
|
||||
litellm_settings:
|
||||
callbacks: ["gcs_bucket"] # 👈 KEY CHANGE
|
|
@ -506,4 +506,4 @@ class FineTuningJobCreate(BaseModel):
|
|||
|
||||
|
||||
class LiteLLMFineTuningJobCreate(FineTuningJobCreate):
|
||||
custom_llm_provider: Literal["openai", "azure"]
|
||||
custom_llm_provider: Literal["openai", "azure", "vertex_ai"]
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue