ray-project / ray-llm

RayLLM - LLMs on Ray
https://aviary.anyscale.com
Apache License 2.0
1.2k stars 87 forks source link

Support for the Mistral based Embeddings models #132

Open lynkz-matt-psaltis opened 5 months ago

lynkz-matt-psaltis commented 5 months ago

For example: https://huggingface.co/intfloat/e5-mistral-7b-instruct

Currently it looks like these models do not support the token_type_ids property which is mandatory in the embedding warmup phase: https://github.com/ray-project/ray-llm/blob/master/rayllm/backend/llm/embedding/embedding_engine.py#L112C14-L112C28

Are you open to PRs on this sort of stuff?