Update optimum_benchmark/scenarios/energy_star/preprocessing_utils.py

huggingface · Nov 19, 2024 · e9b5b32 · e9b5b32
1 parent 82f1ef6
commit e9b5b32
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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(