From 808328e0410b5f90436d9a7cbace6f4e48947fc4 Mon Sep 17 00:00:00 2001 From: Wing Lian Date: Tue, 7 Jan 2025 15:11:07 -0500 Subject: [PATCH] reward can use same batch check --- src/axolotl/prompt_strategies/bradley_terry/chat_template.py | 3 --- 1 file changed, 3 deletions(-) diff --git a/src/axolotl/prompt_strategies/bradley_terry/chat_template.py b/src/axolotl/prompt_strategies/bradley_terry/chat_template.py index 0bbb1de8b..0a02f54e5 100644 --- a/src/axolotl/prompt_strategies/bradley_terry/chat_template.py +++ b/src/axolotl/prompt_strategies/bradley_terry/chat_template.py @@ -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): """