zhangyiss/mlx-examples
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx-examples.git synced 2025-12-16 02:08:55 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity
Files
0cff3b781beb078919111292d7f5a40634d42fdd
mlx-examples/llms/mlx_lm/models/plamo.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

6.5 KiB
Raw Blame History

View Raw
Reference in New Issue View Git Blame Copy Permalink