From 18ed87edc4644edef8c5189ed5cd3c660f1e08d1 Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Fri, 10 May 2024 18:46:25 -0700 Subject: [PATCH] add triton embedding to _init --- litellm/__init__.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/litellm/__init__.py b/litellm/__init__.py index 8d4666a3b02..aedf4213917 100644 --- a/litellm/__init__.py +++ b/litellm/__init__.py @@ -1,5 +1,6 @@ ### Hide pydantic namespace conflict warnings globally ### import warnings + warnings.filterwarnings("ignore", message=".*conflict with protected namespace.*") ### INIT VARIABLES ### import threading, requests, os @@ -537,6 +538,7 @@ provider_list: List = [ "xinference", "fireworks_ai", "watsonx", + "triton", "predibase", "custom", # custom apis ]