From d14c9fe7ea6e7390513ad955d595d3c20fc9ab21 Mon Sep 17 00:00:00 2001 From: Chunyang Wen Date: Sun, 9 Mar 2025 06:51:04 +0800 Subject: [PATCH] Add file info when raising errors in save (#1943) --- python/mlx/nn/layers/base.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/mlx/nn/layers/base.py b/python/mlx/nn/layers/base.py index f24bd1806..3a696df71 100644 --- a/python/mlx/nn/layers/base.py +++ b/python/mlx/nn/layers/base.py @@ -210,7 +210,7 @@ class Module(dict): mx.save_safetensors(file, params_dict) else: raise ValueError( - "Unsupported file extension. Use '.npz' or '.safetensors'." + f"Unsupported file extension for {file}. Use '.npz' or '.safetensors'." ) @staticmethod