zhangyiss/mlx-examples
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx-examples.git synced 2025-06-25 18:11:17 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity
c4833a2f55
mlx-examples/llms/mlx_lm/tuner/__init__.py

3 lines
92 B
Python
Raw Normal View History

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 21:38:42 +08:00
from .trainer import TrainingArgs, evaluate, train
from .utils import linear_to_lora_layers
Reference in New Issue Copy Permalink