{ | |
"adapter_path": "/Users/hebertg/iOSDev/KnowledgeBase/MLX-Fine-Tuning-Pipeline/outputs/knowledgebase-adapter", | |
"batch_size": 2, | |
"config": null, | |
"data": "/Users/hebertg/iOSDev/KnowledgeBase/MLX-Fine-Tuning-Pipeline/outputs/mlx_data", | |
"fine_tune_type": "lora", | |
"grad_checkpoint": false, | |
"iters": 100, | |
"learning_rate": 0.0001, | |
"lora_parameters": { | |
"rank": 8, | |
"dropout": 0.0, | |
"scale": 20.0 | |
}, | |
"lr_schedule": null, | |
"mask_prompt": false, | |
"max_seq_length": 512, | |
"model": "mlx-community/Qwen2.5-1.5B-Instruct-4bit", | |
"num_layers": 16, | |
"optimizer": "adam", | |
"optimizer_config": { | |
"adam": {}, | |
"adamw": {} | |
}, | |
"resume_adapter_file": null, | |
"save_every": 50, | |
"seed": 0, | |
"steps_per_eval": 25, | |
"steps_per_report": 10, | |
"test": false, | |
"test_batches": 500, | |
"train": true, | |
"val_batches": 25, | |
"wandb": null | |
} |