mirror of
https://github.com/ml-explore/mlx-examples.git
synced 2025-07-13 05:01:12 +08:00
c37e26a1a3
* Add pfnet/plamo-2-1b * Fix cache.py to support non-top level layers * Use mlx's BaseModelArgs * Fix model * Use sanitize() * Remove unnecessary changes * Add plamo2.py * Apply formatter * Fix some part * Allow a cache obj defined externally * Fix channel first weights to channel last for right use of MLX's conv1d * Remove unused code part * Give all inputs when it's the first time call of model * Fix import * Include .jsonl files to download from Huggingface hub * Fix reference to layers * Remove unnecessary code and add a test for plamo2 * Do not pass mask to prepare_inputs_for_generation * Fix to use repeat instead of tile * Add state property to PlamoCache * Add __iter__ and __next__ methods to PlamoCache * cleanup * cleanup * fix --------- Co-authored-by: Awni Hannun <awni.hannun@gmail.com> |
||
|---|---|---|
| .. | ||
| examples | ||
| models | ||
| tuner | ||
| __init__.py | ||
| _version.py | ||
| cache_prompt.py | ||
| chat.py | ||
| convert.py | ||
| evaluate.py | ||
| fuse.py | ||
| generate.py | ||
| gguf.py | ||
| LORA.md | ||
| lora.py | ||
| MANAGE.md | ||
| manage.py | ||
| MERGE.md | ||
| merge.py | ||
| py.typed | ||
| README.md | ||
| requirements.txt | ||
| sample_utils.py | ||
| SERVER.md | ||
| server.py | ||
| tokenizer_utils.py | ||
| UPLOAD.md | ||
| utils.py | ||
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.