Files
mlx-examples/llms/mlx_lm/models/phi3small.py
Alex Barron d4ef909d4a Length masking for batch inputs (#1173)
* length masking

* add mask to mlx_lm model interface

* remove lengths

* fix test:

* comment + fix
2024-12-18 19:43:52 -08:00

10 KiB