diff --git a/src/levanter/utils/flop_utils.py b/src/levanter/utils/flop_utils.py index 27278ca08..7d5c4fc0a 100644 --- 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, },