mlx-examples/llms/mlx_lm at e5c98f471594588039ea63786b57c12ed9ca3251 - mlx-examples - Gitea for Geophysics

zhangyiss/mlx-examples

mirror of https://github.com/ml-explore/mlx-examples.git synced 2025-12-16 02:08:55 +08:00

Files

History

mark e5c98f4715 Update utils.py

Change to enable saving the kv-cache as a safetensors file after a text completion; after generate step has finished creating all the tokens, the key values cache is made into a dict and saved using mx.save_safetensors to a user-specified file location; similar to cache_prompt.

2024-09-26 16:58:02 +01:00

..

Example of response generation with optional arguments (#853 )

2024-07-09 06:49:59 -07:00

feat(mlx_lm): Nemotron (#949 )

2024-08-29 21:08:57 -07:00

feat(mlx_lm): Nemotron (#949 )

2024-08-29 21:08:57 -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

Update LLM generation docs to use chat template (#973 )

2024-09-07 06:06:15 -07:00

cache_prompt.py

Fix the cache_prompt (#979 )

2024-09-06 20:19:27 -07:00

convert.py

Create executables for generate, lora, server, merge, convert (#682 )

2024-04-16 16:08:49 -07:00

fuse.py

Allow the entire model to be targed for LoRA and DoRA fine tuning: LoRA and DoRA embeddings with small DoRALinear bug fix (#914 )

2024-08-16 07:38:36 -07:00

generate.py

Add prompt piping (#962 )

2024-09-03 13:29:10 -07:00

gguf.py

Fix export to gguf (#993 )

2024-09-20 13:33:45 -07:00

LORA.md

Fix bug in upload + docs nit (#981 )

2024-09-07 14:46:57 -07:00

lora.py

Pass use_dora parameter to linear_to_lora_layers (#885 )

2024-07-11 14:34:34 -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

Adapters loading (#902 )

2024-08-01 16:18:18 -07:00

server.py

Predict stop sequence matches during streaming (#541 )

2024-08-06 15:24:15 -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

Update utils.py

2024-09-26 16:58:02 +01:00

README.md

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.