phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

all fixes to linting

Browse source
This commit is contained in:
ishaan-jaff 2023-08-18 11:56:44 -07:00
parent 5e7d22512d
commit 1bb2aefea1
8 changed files with 44 additions and 34 deletions
Download patch file Download diff file Expand all files Collapse all files

9
litellm/llms/huggingface_restapi.py
View file

@ -45,10 +45,11 @@ class HuggingfaceRestAPILLM:
litellm_params=None,
logger_fn=None,
): # logic for parsing in - calling - parsing out model completion calls
completion_url: str = ""
if custom_api_base:
completion_url: Optional[str] = custom_api_base
completion_url = custom_api_base
elif "HF_API_BASE" in os.environ:
completion_url = os.getenv("HF_API_BASE")
completion_url = os.getenv("HF_API_BASE", "")
else:
completion_url = f"https://api-inference.huggingface.co/models/{model}"
prompt = ""
@ -137,5 +138,7 @@ class HuggingfaceRestAPILLM:
return model_response
pass
def embedding(self): # logic for parsing in - calling - parsing out model embedding calls
def embedding(
self,
): # logic for parsing in - calling - parsing out model embedding calls
pass