Let reward model take text inputs instead of message lists (#1907)

Co-authored-by: Kyle Corbitt <kyle@corbt.com>
2024-11-03 13:27:12 -08:00
parent 793b79dbe9
commit 2ce32db6fb
12 changed files with 43 additions and 58 deletions
--- a/test/srt/run_suite.py
+++ b/test/srt/run_suite.py
@@ -8,7 +8,7 @@ suites = {
        "models/test_embedding_models.py",
        "models/test_generation_models.py",
        "models/test_lora.py",
-        # "models/test_reward_models.py",
+        "models/test_reward_models.py",
        "sampling/penaltylib",
        "test_chunked_prefill.py",
        "test_double_sparsity.py",