support internlm2-reward (#1994)

2024-11-12 07:09:58 +08:00
parent 3e33574374
commit f18b9c7252
3 changed files with 65 additions and 0 deletions
--- a/docs/references/supported_models.md
+++ b/docs/references/supported_models.md
@@ -42,6 +42,8 @@
  - `python -m sglang.launch_server --model-path Skywork/Skywork-Reward-Llama-3.1-8B-v0.2 --is-embedding`
 - Gemma2ForSequenceClassification
  - `python -m sglang.launch_server --model-path Skywork/Skywork-Reward-Gemma-2-27B-v0.2 --is-embedding`
+- InternLM2ForRewardModel
+  - `python -m sglang.launch_server --model-path internlm/internlm2-7b-reward --is-embedding --trust-remote-code`

 ## How to Support a New Model