Skip to content

✨ Add feature extraction task for Nebius #1436

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
merged 6 commits into from
May 9, 2025

Conversation

diadorer
Copy link
Contributor

@diadorer diadorer commented May 6, 2025

Added support for embedding models for Nebius provider.

Copy link
Contributor

@hanouticelina hanouticelina left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Looks good! thanks @diadorer for the contribution, exciting to have Nebius supporting feature-extraction task 🚀 I tested the PR and it works as expected!

could you update the mapping to add the supported models (let's add them in staging status for now)? If i'm not mistaken, Nebius supports BAAI/bge-multilingual-gemma2 , intfloat/e5-mistral-7b-instruct and BAAI/bge-en-icl.

you can find the mapping API documentation here: https://huggingface.co/docs/inference-providers/register-as-a-provider#3-model-mapping-api

@diadorer
Copy link
Contributor Author

diadorer commented May 7, 2025

Thanks for the review @hanouticelina!
These models are already there with a staging status. I guess only org members can see them.

@hanouticelina
Copy link
Contributor

@diadorer Ah yes my bad, I was not SSO-logged. thank you!

@diadorer
Copy link
Contributor Author

diadorer commented May 9, 2025

@hanouticelina could we merge it?

@hanouticelina
Copy link
Contributor

Let's merge this 🚀 thanks again @diadorer for the PR!

@hanouticelina hanouticelina merged commit a4e8f37 into huggingface:main May 9, 2025
4 checks passed
@diadorer
Copy link
Contributor Author

diadorer commented May 9, 2025

Thanks you @hanouticelina!

May I ask you to have a look at Python's one as well?
huggingface/huggingface_hub#3057

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants