added 1xA100 (#202)

huggingface · May 16, 2024 · 9b28308 · 9b28308
1 parent 6351381
commit 9b28308
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/llm_perf/update_llm_perf_leaderboard.py b/llm_perf/update_llm_perf_leaderboard.py
@@ -30,5 +30,8 @@ def gather_benchmarks(subset: str, machine: str):
 
 
 for subset in ["unquantized", "bnb", "awq", "gptq"]:
-    for machine in ["1xA10"]:
-        gather_benchmarks(subset, machine)
+    for machine in ["1xA10", "1xA100"]:
+        try:
+            gather_benchmarks(subset, machine)
+        except Exception:
+            print(f"Subset {subset} for machine {machine} not found")