mlx-examples/llms/mlx_lm
Anchen 362e88a744
feat: move lora into mlx-lm (#337)
* feat: Add lora and qlora training to mlx-lm


---------

Co-authored-by: Awni Hannun <awni@apple.com>
2024-01-23 08:44:37 -08:00
..
models feat: move lora into mlx-lm (#337) 2024-01-23 08:44:37 -08:00
tuner feat: move lora into mlx-lm (#337) 2024-01-23 08:44:37 -08:00
__init__.py Mlx llm package (#301) 2024-01-12 10:25:56 -08:00
convert.py feat: move lora into mlx-lm (#337) 2024-01-23 08:44:37 -08:00
fuse.py feat: move lora into mlx-lm (#337) 2024-01-23 08:44:37 -08:00
generate.py Add colorized output option to generate script (#347) 2024-01-23 05:25:44 -08:00
LORA.md feat: move lora into mlx-lm (#337) 2024-01-23 08:44:37 -08:00
lora.py feat: move lora into mlx-lm (#337) 2024-01-23 08:44:37 -08:00
README.md feat: move lora into mlx-lm (#337) 2024-01-23 08:44:37 -08:00
requirements.txt refactor(qwen): moving qwen into mlx-lm (#312) 2024-01-22 15:00:07 -08:00
UPLOAD.md Mlx llm package (#301) 2024-01-12 10:25:56 -08:00
utils.py feat: move lora into mlx-lm (#337) 2024-01-23 08:44:37 -08:00

Generate Text with MLX and 🤗 Hugging Face

This an example of large language model text generation that can pull models from the Hugging Face Hub.

For more information on this example, see the README in the parent directory.

This package also supports fine tuning with LoRA or QLoRA. For more information see the LoRA documentation.