Reranker detected as embedder Jina rerank tiny

[rawsh-rubrik]

trafficstars

System Info

infinity onnx image latest

Information

• [ ] Docker + cli
• [ ] pip + cli
• [ ] pip + usage of Python interface

Tasks

• [ ] An officially supported CLI command
• [ ] My own modifications

Reproduction

jinaai/jina-reranker-v1-tiny-en tensorrt device optimum engine

{"data":[{"id":"jinaai/jina-reranker-v1-tiny-en","stats":{"queue_fraction":0.0,"queue_absolute":0,"results_pending":0,"batch_size":32},"object":"model","owned_by":"infinity","created":1734462700,"backend":"optimum","capabilities":["embed"]}],"object":"list"}

[michaelfeil]

https://huggingface.co/jinaai/jina-reranker-v1-tiny-en/discussions/9

Please make the jina team aware of this! @wirthual Already has a PR ready, which currently does not work and needs common resolution.

the model needs to be named “forsequenceclassification” to be detected as sequence classification model.

[rawsh-rubrik]

@michaelfeil gotcha thanks! will fork for now

[michaelfeil]

The name of the model also needs to resolve in the config.json - might need a try or two.

once you got a fork working, use —revision for infinity.