mlx-examples/llms/mlx_lm
Awni Hannun fca087be49
More cache improvements (#1015)
* fix rotating kv cache for chat use case

* reorg + fixes to caching, unify prompt caching across types and use cases for e.g. caching during a chat

* nit in chat

* fix tests

* fix tests

* fix tests

* docs

* chat command

* comments + docs

* Define meta_state on all Cache implementations

* fixes + trim_prompt_cache api

* fix default model

---------

Co-authored-by: Angelos Katharopoulos <a_katharopoulos@apple.com>
2024-10-07 20:45:51 -07:00
..
examples More cache improvements (#1015) 2024-10-07 20:45:51 -07:00
models More cache improvements (#1015) 2024-10-07 20:45:51 -07:00
tuner Feature: QDoRA (#891) 2024-09-30 08:01:11 -07:00
__init__.py Make sure to import the correct "version" module when installing mlx_whisper and mlx_lm from local source code. (#969) 2024-09-03 13:16:21 -07:00
_version.py More cache improvements (#1015) 2024-10-07 20:45:51 -07:00
cache_prompt.py More cache improvements (#1015) 2024-10-07 20:45:51 -07:00
chat.py More cache improvements (#1015) 2024-10-07 20:45:51 -07:00
convert.py Create executables for generate, lora, server, merge, convert (#682) 2024-04-16 16:08:49 -07:00
fuse.py Adding full finetuning (#903) 2024-09-29 17:12:47 -07:00
generate.py More cache improvements (#1015) 2024-10-07 20:45:51 -07:00
gguf.py Fix export to gguf (#993) 2024-09-20 13:33:45 -07:00
LORA.md LoRA: Support HuggingFace dataset via data parameter (#996) 2024-09-30 07:36:21 -07:00
lora.py LoRA: Support HuggingFace dataset via data parameter (#996) 2024-09-30 07:36:21 -07:00
MANAGE.md Add model management functionality for local caches (#736) 2024-05-03 12:20:13 -07:00
manage.py Add model management functionality for local caches (#736) 2024-05-03 12:20:13 -07:00
MERGE.md Create executables for generate, lora, server, merge, convert (#682) 2024-04-16 16:08:49 -07:00
merge.py Create executables for generate, lora, server, merge, convert (#682) 2024-04-16 16:08:49 -07:00
py.typed Add py.typed to support PEP-561 (type-hinting) (#389) 2024-01-30 21:17:38 -08:00
README.md feat: move lora into mlx-lm (#337) 2024-01-23 08:44:37 -08:00
requirements.txt Use fast rope (#945) 2024-08-23 13:18:51 -07:00
sample_utils.py Min P implementation (#926) 2024-08-15 15:45:02 -07:00
SERVER.md Add /v1/models endpoint to mlx_lm.server (#984) 2024-09-28 07:21:11 -07:00
server.py Server: support function calling (#1003) 2024-10-02 12:36:07 -07:00
tokenizer_utils.py Fix setattr for the TokenizerWrapper (#961) 2024-08-28 14:47:33 -07:00
UPLOAD.md Mlx llm package (#301) 2024-01-12 10:25:56 -08:00
utils.py More cache improvements (#1015) 2024-10-07 20:45:51 -07: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.