fix device kind for mfu v5e (#713)

stanford-crfm · Aug 29, 2024 · ffa8e28 · ffa8e28
1 parent 4913df2
commit ffa8e28
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/levanter/utils/flop_utils.py b/src/levanter/utils/flop_utils.py
@@ -138,7 +138,7 @@ def lm_flops_per_token(
         "int8": 275e12,
     },
     # Source: https://cloud.google.com/tpu/docs/v5e
-    "TPU v5 lite": {
+    "tpu v5 lite": {
         "bf16": 197e12,
         "int8": 393e12,
     },