├── .github └── workflows │ └── python-app.yml ├── CHANGELOG.md ├── LICENSE ├── README.md ├── constraints.txt ├── pyproject.toml ├── requirements.txt └── src └── mlx_tuning_fork ├── __init__.py ├── config.py ├── dataset.py ├── generate.py ├── prompt_templates ├── __init__.py ├── alpaca.py ├── chatml.py ├── gemma.py ├── llama3.py ├── mistral.py └── phi.py ├── reporting.py ├── training.py ├── tuning ├── __init__.py ├── input_masking.py └── utils.py └── wandb_sweep.py /.github/workflows/python-app.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chimezie/mlx-tuning-fork/HEAD/.github/workflows/python-app.yml -------------------------------------------------------------------------------- /CHANGELOG.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chimezie/mlx-tuning-fork/HEAD/CHANGELOG.md -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chimezie/mlx-tuning-fork/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chimezie/mlx-tuning-fork/HEAD/README.md -------------------------------------------------------------------------------- /constraints.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chimezie/mlx-tuning-fork/HEAD/constraints.txt -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chimezie/mlx-tuning-fork/HEAD/pyproject.toml -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- 1 | mlx 2 | click 3 | tqdm 4 | ogbujipt 5 | numpy 6 | mlx-lm 7 | pyyaml 8 | pytest -------------------------------------------------------------------------------- /src/mlx_tuning_fork/__init__.py: -------------------------------------------------------------------------------- 1 | __version__ = "0.4.0" -------------------------------------------------------------------------------- /src/mlx_tuning_fork/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chimezie/mlx-tuning-fork/HEAD/src/mlx_tuning_fork/config.py -------------------------------------------------------------------------------- /src/mlx_tuning_fork/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chimezie/mlx-tuning-fork/HEAD/src/mlx_tuning_fork/dataset.py -------------------------------------------------------------------------------- /src/mlx_tuning_fork/generate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chimezie/mlx-tuning-fork/HEAD/src/mlx_tuning_fork/generate.py -------------------------------------------------------------------------------- /src/mlx_tuning_fork/prompt_templates/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/mlx_tuning_fork/prompt_templates/alpaca.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chimezie/mlx-tuning-fork/HEAD/src/mlx_tuning_fork/prompt_templates/alpaca.py -------------------------------------------------------------------------------- /src/mlx_tuning_fork/prompt_templates/chatml.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chimezie/mlx-tuning-fork/HEAD/src/mlx_tuning_fork/prompt_templates/chatml.py -------------------------------------------------------------------------------- /src/mlx_tuning_fork/prompt_templates/gemma.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chimezie/mlx-tuning-fork/HEAD/src/mlx_tuning_fork/prompt_templates/gemma.py -------------------------------------------------------------------------------- /src/mlx_tuning_fork/prompt_templates/llama3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chimezie/mlx-tuning-fork/HEAD/src/mlx_tuning_fork/prompt_templates/llama3.py -------------------------------------------------------------------------------- /src/mlx_tuning_fork/prompt_templates/mistral.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chimezie/mlx-tuning-fork/HEAD/src/mlx_tuning_fork/prompt_templates/mistral.py -------------------------------------------------------------------------------- /src/mlx_tuning_fork/prompt_templates/phi.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chimezie/mlx-tuning-fork/HEAD/src/mlx_tuning_fork/prompt_templates/phi.py -------------------------------------------------------------------------------- /src/mlx_tuning_fork/reporting.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chimezie/mlx-tuning-fork/HEAD/src/mlx_tuning_fork/reporting.py -------------------------------------------------------------------------------- /src/mlx_tuning_fork/training.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chimezie/mlx-tuning-fork/HEAD/src/mlx_tuning_fork/training.py -------------------------------------------------------------------------------- /src/mlx_tuning_fork/tuning/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/mlx_tuning_fork/tuning/input_masking.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chimezie/mlx-tuning-fork/HEAD/src/mlx_tuning_fork/tuning/input_masking.py -------------------------------------------------------------------------------- /src/mlx_tuning_fork/tuning/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chimezie/mlx-tuning-fork/HEAD/src/mlx_tuning_fork/tuning/utils.py -------------------------------------------------------------------------------- /src/mlx_tuning_fork/wandb_sweep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chimezie/mlx-tuning-fork/HEAD/src/mlx_tuning_fork/wandb_sweep.py --------------------------------------------------------------------------------