Gemma2 reward model support (#1954)

2024-11-08 14:42:27 +08:00
parent 5bc2508b80
commit f16eb15d0d
2 changed files with 105 additions and 1 deletions
--- a/docs/references/supported_models.md
+++ b/docs/references/supported_models.md
@@ -40,7 +40,8 @@

 - LlamaForSequenceClassification
  - `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`

 ## How to Support a New Model