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()
|
# 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