zhangyiss/mlx-examples
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx-examples.git synced 2025-08-29 18:26:37 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix synch

Browse Source
This commit is contained in:
Awni Hannun 2024-10-24 06:19:19 -07:00
parent 131ccbe6df
commit 2822acc98a
1 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
llms/mlx_lm/utils.py
View File

@ -58,8 +58,7 @@ def wired_limit(model):
try:
yield None
finally:
# TODO... expose a synchronize??
mx.zeros((1,)).item()
mx.synchronize()
mx.metal.set_wired_limit(old_limit)