-
Notifications
You must be signed in to change notification settings - Fork 372
✨ 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
✨ Add feature extraction task for Nebius #1436
Conversation
There was a problem hiding this 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
Thanks for the review @hanouticelina! |
@diadorer Ah yes my bad, I was not SSO-logged. thank you! |
@hanouticelina could we merge it? |
Let's merge this 🚀 thanks again @diadorer for the PR! |
Thanks you @hanouticelina! May I ask you to have a look at Python's one as well? |
Added support for embedding models for Nebius provider.