mirror of
https://github.com/BerriAI/litellm.git
synced 2025-04-25 18:54:30 +00:00
parent
1f17daf52c
commit
c7a04140e9
19 changed files with 141 additions and 191 deletions
|
@ -84,7 +84,8 @@ class BaseLLMEmbeddingTest(ABC):
|
|||
litellm.set_verbose = True
|
||||
from litellm.utils import supports_embedding_image_input
|
||||
|
||||
litellm.model_cost = litellm.get_locally_cached_model_cost_map()
|
||||
os.environ["LITELLM_LOCAL_MODEL_COST_MAP"] = "True"
|
||||
litellm.model_cost = litellm.get_model_cost_map(url="")
|
||||
|
||||
base_embedding_call_args = self.get_base_embedding_call_args()
|
||||
if not supports_embedding_image_input(base_embedding_call_args["model"], None):
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue