diff --git a/configs/stability_3b.yml b/configs/stability_3b.yml index c5f2198d8..ecbd37980 100644 --- a/configs/stability_3b.yml +++ b/configs/stability_3b.yml @@ -20,7 +20,7 @@ lora_fan_in_fan_out: false wandb_project: stable-alpaca-3b wandb_watch: wandb_run_id: -wandb_log_model: checkpoint +wandb_log_model: output_dir: ./stable-alpaca-3b batch_size: 2 micro_batch_size: 1