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

docs(huggingface.md): add text-classification to huggingface docs

Browse source
This commit is contained in:
Krrish Dholakia 2024-05-10 14:39:14 -07:00
parent 50be25d11a
commit d4d175030f
3 changed files with 177 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

3
litellm/tests/test_completion.py
View file

@ -1291,9 +1291,8 @@ def test_hf_classifier_task():
user_message = "I like you. I love you"
messages = [{"content": user_message, "role": "user"}]
response = completion(
model="huggingface/shahrukhx01/question-vs-statement-classifier",
model="huggingface/text-classification/shahrukhx01/question-vs-statement-classifier",
messages=messages,
hf_task="text-classification",
)
print(f"response: {response}")
assert isinstance(response, litellm.ModelResponse)