diff --git a/examples/mpt-7b/config.yml b/examples/mpt-7b/config.yml index f33452266..ca6a0d5c3 100644 --- a/examples/mpt-7b/config.yml +++ b/examples/mpt-7b/config.yml @@ -22,7 +22,7 @@ lora_fan_in_fan_out: false wandb_project: mpt-alpaca-7b wandb_watch: wandb_run_id: -wandb_log_model: checkpoint +wandb_log_model: output_dir: ./mpt-alpaca-7b batch_size: 1 micro_batch_size: 1