fix bugs

huggingface · Oct 14, 2024 · 49a5f70 · 49a5f70
1 parent 755c19d
commit 49a5f70
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/optimum_benchmark/scenarios/energy_star/scenario.py b/optimum_benchmark/scenarios/energy_star/scenario.py
@@ -94,7 +94,9 @@ def run(self, backend: Backend[BackendConfigT]) -> BenchmarkReport:
                 )
             self.config.input_shapes["batch_size"] //= torch.distributed.get_world_size()
 
-        self.energy_tracker = EnergyTracker(device=backend.config.device, device_ids=backend.config.device_ids, backend=backend.NAME)
+        self.energy_tracker = EnergyTracker(
+            device=backend.config.device, device_ids=backend.config.device_ids, backend=backend.NAME
+        )
 
         LOGGER.info("\t+ Loading dataset")
         raw_dataset = load_dataset(