mlx-examples/llms/mlx_lm/requirements.txt
Awni Hannun 2146bcd7ee
Quantize embedding / Update quantize API (#680)
* more async eval

* quantize embedding / update quantize api

* more updates for quantize

* update for quantize embeddings

* update sd quant API

* update sdxl quants

* error for datasets < batch_size

* async

* fix config loading

* fix quant

* fix tests

* fix req

* remove lm head if tie weights is true

* fix test
2024-04-18 18:16:10 -07:00

7 lines
60 B
Plaintext

mlx>=0.11
numpy
transformers>=4.39.3
protobuf
pyyaml
jinja2