mirror of
https://github.com/ml-explore/mlx-examples.git
synced 2025-09-10 21:37:45 +08:00
Generalize prompt_feature and completion_feature for use in local datasets to facilitate compatibility with many other training dataset formats.
This commit is contained in:

committed by
Awni Hannun

parent
bf2da36fc6
commit
db9898d104
@@ -61,6 +61,8 @@ CONFIG_DEFAULTS = {
|
||||
"config": None,
|
||||
"grad_checkpoint": False,
|
||||
"lr_schedule": None,
|
||||
"prompt_feature": "prompt",
|
||||
"completion_feature": "completion",
|
||||
"lora_parameters": {"rank": 8, "alpha": 16, "dropout": 0.0, "scale": 10.0},
|
||||
}
|
||||
|
||||
|
Reference in New Issue
Block a user