remove energy tracking

huggingface · Feb 12, 2024 · 47f7df3 · 47f7df3
1 parent 1025d91
commit 47f7df3
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/test_api.py b/tests/test_api.py
@@ -44,8 +44,8 @@
     ("transformers", "semantic-segmentation", "google/vit-base-patch16-224"),
 ]
 BENCHMARK_CONFIGS = [
-    InferenceConfig(latency=True, memory=True, energy=True),
-    TrainingConfig(latency=True, memory=True, energy=True),
+    InferenceConfig(latency=True, memory=True),
+    TrainingConfig(latency=True, memory=True),
 ]
 LAUNCHER_CONFIGS = [
     TorchrunConfig(nproc_per_node=2, device_isolation=False),