reward can use same batch check

This commit is contained in:
Wing Lian
2025-01-07 15:11:07 -05:00
parent 28ef5e8d5a
commit ec4dfb02c8

View File

@@ -21,9 +21,6 @@ class BTChatTemplateStrategy(ChatTemplateStrategy):
Bradley-Terry reward model pairwise chat template prompt strategy.
"""
def is_prompt_batched(self, prompt: dict[str, Any]) -> bool:
return all(isinstance(v, list) for v in prompt.values())
def _tokenize_single_prompt(self, prompt):
"""