diff --git a/esm/README.md b/esm/README.md index 549331ef..5cda12cc 100644 --- a/esm/README.md +++ b/esm/README.md @@ -111,7 +111,7 @@ Benchmark PyTorch MPS performance: python benchmarks/benchmark_pt.py ``` -Expected performance on M4 MacBook Pro (batch_size = 5): +Expected performance on M4 MacBook Pro (ESM-2 650M, batch_size = 5): - MLX: 299 ms per step, 16.71 sequences/sec - PyTorch MPS: 402 ms per step, 12.43 sequences/sec