Skip to content

Support for deberta-v2 model variant #199

@w3iw3i

Description

@w3iw3i

Feature request

When attempting to use this reranker model mxbai-rerank-large-v1 from huggingface with TEI, I got the following error message:

Error: Could not create backend

Caused by:
    Could not start backend: Model is not supported: unknown variant `deberta-v2`, expected one of `bert`, `xlm-roberta`, `camembert`, `roberta`, `distilbert`, `nomic_bert` at line 21 column 28

Steps to reproduce

model=mixedbread-ai/mxbai-rerank-large-v1
volume=$PWD/models

docker run -p 8088:80 -v $volume:/models --pull always ghcr.io/huggingface/text-embeddings-inference:cpu-1.1.0 --model-id $model

Motivation

Support for deberta-v2 model variant would allow the usage of mxbai-rerank-large-v1 model with /rerank endpoint

Your contribution

Support for deberta-v2 model variant

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions