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

don't throw if no metal

Browse Source
This commit is contained in:
Awni Hannun 2024-10-09 19:13:33 -07:00
parent d6222ae7ff
commit 1b05b51dc5
1 changed files with 2 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
llms/mlx_lm/server.py
View File

@ -31,10 +31,8 @@ from .utils import generate_step, load
def get_system_fingerprint():
return (
f"{__version__}-{mx.__version__}-{platform.platform()}-"
f"{mx.metal.device_info().get('architecture', '')}"
)
gpu_arch = mx.metal.device_info()["architecture"] if mx.metal.is_available() else ""
return f"{__version__}-{mx.__version__}-{platform.platform()}-{gpu_arch}"
class StopCondition(NamedTuple):