Update lora_config.yaml with new param: num_layers (#1068)

ml-explore · Oct 26, 2024 · ab4bf05 · ab4bf05
1 parent 4971462
commit ab4bf05
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/llms/mlx_lm/examples/lora_config.yaml b/llms/mlx_lm/examples/lora_config.yaml
@@ -14,7 +14,7 @@ data: "/path/to/training/data"
 seed: 0
 
 # Number of layers to fine-tune
-lora_layers: 16
+num_layers: 16
 
 # Minibatch size.
 batch_size: 4