mlx-examples/llms/mlx_lm/tuner/__init__.py
madroid c457a3f88b
LoRA: Extract small function (#614)
* LoRA: Extract pre_processing_model  function

* LoRA: Extract small functions(train_model,evaluate_model)

* move test case to test_tuner_utils.py

* nits

* nits

* remove extra param, validate at it 0

* version

* fix test

---------

Co-authored-by: Awni Hannun <awni@apple.com>
2024-06-02 06:38:42 -07:00

3 lines
92 B
Python

from .trainer import TrainingArgs, evaluate, train
from .utils import linear_to_lora_layers