Update README.md

This commit is contained in:
Vincent Amato
2025-08-16 11:08:58 -04:00
parent 93b2a8fa4a
commit afe88cddde

View File

@@ -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