Add Reward API Docs etc (#1910)

Co-authored-by: Chayenne <zhaochenyang@g.ucla.edu>
2024-11-03 22:33:03 -08:00
parent 1853c3523b
commit 704f8e8ed1
6 changed files with 90 additions and 16 deletions
--- a/test/srt/models/test_reward_models.py
+++ b/test/srt/models/test_reward_models.py
@@ -22,6 +22,7 @@ from sglang.test.runners import HFRunner, SRTRunner

 MODELS = [
    ("LxzGordon/URM-LLaMa-3.1-8B", 1, 3e-2),
+    ("Skywork/Skywork-Reward-Llama-3.1-8B-v0.2", 1, 3e-2),
 ]
 TORCH_DTYPES = [torch.float16]