huggingface / huggingface_hub

The official Python client for the Huggingface Hub.
https://huggingface.co/docs/huggingface_hub
Apache License 2.0
1.82k stars 470 forks source link

Add `prompt_name` to feature-extraction + update types #2363

Closed Wauplin closed 1 day ago

Wauplin commented 2 days ago

Related to https://github.com/huggingface/huggingface.js/pull/781.

This PR adds prompt_name to InferenceClient.feature_extraction which was introduced in https://github.com/huggingface/text-embeddings-inference/pull/312. I also took the opportunity to add truncation_direction.

Wauplin commented 1 day ago

Thanks for the review!