From 6f8765125b28289e16bbeb890899e527f16e3480 Mon Sep 17 00:00:00 2001 From: ishaan-jaff Date: Mon, 4 Dec 2023 11:57:41 -0800 Subject: [PATCH] (docs) embedding: add api_base for HF --- docs/my-website/docs/embedding/supported_embedding.md | 11 +++++++++++ 1 file changed, 11 insertions(+) diff --git a/docs/my-website/docs/embedding/supported_embedding.md b/docs/my-website/docs/embedding/supported_embedding.md index 009e5337b..3c13c5732 100644 --- a/docs/my-website/docs/embedding/supported_embedding.md +++ b/docs/my-website/docs/embedding/supported_embedding.md @@ -182,6 +182,17 @@ response = embedding( input=["good morning from litellm"] ) ``` +### Usage - Custom API Base +```python +from litellm import embedding +import os +os.environ['HUGGINGFACE_API_KEY'] = "" +response = embedding( + model='huggingface/microsoft/codebert-base', + input=["good morning from litellm"], + api_base = "https://p69xlsj6rpno5drq.us-east-1.aws.endpoints.huggingface.cloud" +) +``` | Model Name | Function Call | Required OS Variables | |-----------------------|--------------------------------------------------------------|-------------------------------------------------|