zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-12-16 01:49:05 +08:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

improve quant docs

Browse Source
This commit is contained in:
Awni Hannun
2025-10-28 07:12:35 -07:00
parent 94fe5114fa
commit 5a043fd793
3 changed files with 42 additions and 32 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
python/tests/test_quantized.py
View File

@@ -146,7 +146,7 @@ class TestQuantized(mlx_tests.MLXTestCase):
w_q, scales = mx.quantize(w, mode="nvfp4")
with self.assertRaises(ValueError):
mx.dequantize(w_q, scales, mode="nvfp4")
mx.dequantize(w_q, scales, bits=3, mode="nvfp4")
with self.assertRaises(ValueError):
mx.dequantize(w_q, scales, group_size=32, mode="nvfp4")