chore: replace deprecated mx.metal.* with mx.set_* API #5

New issue

Closed

opened 2026-02-24 13:42:50 +00:00 by Virgil · 1 comment

Virgil commented

2026-02-24 13:42:50 +00:00

Owner

Summary

MLX has deprecated mx.metal.set_memory_limit and mx.metal.set_cache_limit in favour of:

mx.set_memory_limit
mx.set_cache_limit

The old API will be removed in a future MLX release.

Files to update

scripts/train-4b-lek.py
scripts/train-4b-p0.py
Any other training scripts using mx.metal.*

Change

# Before
mx.metal.set_memory_limit(24 * 1024**3)
mx.metal.set_cache_limit(8 * 1024**3)

# After
mx.set_memory_limit(24 * 1024**3)
mx.set_cache_limit(8 * 1024**3)

Also check mx.get_peak_memory (may already be correct).

## Summary MLX has deprecated `mx.metal.set_memory_limit` and `mx.metal.set_cache_limit` in favour of: - `mx.set_memory_limit` - `mx.set_cache_limit` The old API will be removed in a future MLX release. ## Files to update - `scripts/train-4b-lek.py` - `scripts/train-4b-p0.py` - Any other training scripts using `mx.metal.*` ## Change ```python # Before mx.metal.set_memory_limit(24 * 1024**3) mx.metal.set_cache_limit(8 * 1024**3) # After mx.set_memory_limit(24 * 1024**3) mx.set_cache_limit(8 * 1024**3) ``` Also check `mx.get_peak_memory` (may already be correct).

Charon commented

2026-02-24 18:31:04 +00:00

Owner

No mx.metal.* calls found anywhere in the repo. The referenced scripts (train-4b-lek.py, train-4b-p0.py) are not committed — these are local training scripts on the M3. The deprecated API calls exist only in Virgil's local environment, not in tracked code.

Closing as not applicable to the repo.

No `mx.metal.*` calls found anywhere in the repo. The referenced scripts (`train-4b-lek.py`, `train-4b-p0.py`) are not committed — these are local training scripts on the M3. The deprecated API calls exist only in Virgil's local environment, not in tracked code. Closing as not applicable to the repo.