From e9b5b329295d3403be17060c1789d95b3106d2e8 Mon Sep 17 00:00:00 2001
From: Ilyas Moutawwakil <57442720+IlyasMoutawwakil@users.noreply.github.com>
Date: Tue, 19 Nov 2024 12:21:30 +0100
Subject: [PATCH] Update
 optimum_benchmark/scenarios/energy_star/preprocessing_utils.py

---
 optimum_benchmark/scenarios/energy_star/preprocessing_utils.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/optimum_benchmark/scenarios/energy_star/preprocessing_utils.py b/optimum_benchmark/scenarios/energy_star/preprocessing_utils.py
index afba75e4..d8cfb6a2 100644
--- a/optimum_benchmark/scenarios/energy_star/preprocessing_utils.py
+++ b/optimum_benchmark/scenarios/energy_star/preprocessing_utils.py
@@ -90,7 +90,7 @@ def summarization_preprocessing(
     if getattr(tokenizer, "pad_token", None) is None:
         tokenizer.pad_token = tokenizer.eos_token
 
-    padding = False if config.input_shapes["batch_size"] == 1 else True
+    padding = config.input_shapes["batch_size"] != 1
 
     def tokenize_function(examples):
         return tokenizer(