[Bug]: math-shepherd-mistral-7b-prm does not support the 'embedding' task. Supported tasks: {'generate'}

[KADCA21]

Your current environment

I thought math-shepherd-mistral-7b-prm has already supported 'embedding' task according to this PR

Model Input Dumps

No response

🐛 Describe the bug

Any way to use math-shepherd-mistral-7b-prm in embedding mode?

Before submitting a new issue...

• Make sure you already searched for relevant issues, and asked the chatbot living at the bottom right corner of the documentation page, which can answer lots of frequently asked questions.

[DarkLight1337]

Which version of vLLM are you using? This is not in the latest release, so you have to use vLLM from main branch.