forked from phoenix/litellm-mirror
hf embedding tests
This commit is contained in:
parent
3fbad7dfa7
commit
6e48d92891
1 changed files with 14 additions and 0 deletions
|
@ -33,3 +33,17 @@ def test_cohere_embedding():
|
|||
|
||||
# test_cohere_embedding()
|
||||
|
||||
# comment out hf tests - since hf endpoints are unstable
|
||||
# def test_hf_embedding():
|
||||
# try:
|
||||
# # huggingface/microsoft/codebert-base
|
||||
# # huggingface/facebook/bart-large
|
||||
# response = embedding(
|
||||
# model="huggingface/BAAI/bge-large-zh", input=["good morning from litellm", "this is another item"]
|
||||
# )
|
||||
# print(f"response:", response)
|
||||
# except Exception as e:
|
||||
# pytest.fail(f"Error occurred: {e}")
|
||||
# test_hf_embedding()
|
||||
|
||||
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue