mlx-examples/llms/mlx_lm/requirements.txt

mlx>=0.6
numpy
transformers>=4.38.0
protobuf
pyyaml
jinja2
[mlx-lm] Use sdpa in llama / mistral model (#515) * use sdpa * update a few more models * version * fix stablelm type 2024-03-08 09:41:23 +08:00			`mlx>=0.6`
Support Hugging Face models (#215) * support hf direct models 2024-01-04 07:13:26 +08:00			`numpy`
Gemma support (#474) * gemma support * format * lora support for gemma 2024-02-22 00:47:13 +08:00			`transformers>=4.38.0`
Mlx llm package (#301) * fix converter * add recursive files * remove gitignore * remove gitignore * add packages properly * read me update * remove dup readme * relative * fix convert * fix community name * fix url * version 2024-01-13 02:25:56 +08:00			`protobuf`
Support for slerp merging models (#455) * support for slerp merging models * docs * update docs * format' 2024-02-20 12:37:15 +08:00			`pyyaml`
Support for OpenAI’s fine-tuning dataset format (#548) * LoRA: move load_dataset to tuner/datasets.py file * LoRA: support OpenAI chat format datasets see https://platform.openai.com/docs/guides/fine-tuning/example-format * LoRA: support OpenAI completion format datasets * LoRA: formatting dataset timing to reduce memory footprint * Refactor dataset item access in PromptCompletionDataset * Update mlx_lm/LORA.md * Update mlx_lm/LORA.md * check Unsupported data format * add tests, fine-tune doc * add tests, fine-tune doc * add jinja2 for chat template * nits in readme * nits in readme --------- Co-authored-by: Awni Hannun <awni@apple.com> 2024-03-20 07:45:46 +08:00			`jinja2`