Files
mlx-examples/llms/tests/test_sample_utils.py
Neil Mehta 4c3df00162 make_sampler creates sampler chain with all sampling parameters (#1330)
* top_p refactor

* top_k and min_p refactor

* Create sampler chain

* Remove unnecessary mx.where

* Use mx.allclose
2025-03-11 13:37:35 -07:00

3.6 KiB