Krrish Dholakia 2024-12-11 01:03:57 -08:00
parent e02c4b8e9d
commit fd97b9d966
8 changed files with 197 additions and 62 deletions

View file

@ -6373,6 +6373,8 @@ class ProviderConfigManager:
return litellm.OllamaConfig()
elif litellm.LlmProviders.PREDIBASE == provider:
return litellm.PredibaseConfig()
elif litellm.LlmProviders.TRITON == provider:
return litellm.TritonConfig()
return litellm.OpenAIGPTConfig()