diff --git a/configs/cerebras_1_3B_alpaca.yml b/configs/cerebras_1_3B_alpaca.yml index d749ca72b..2c5534552 100644 --- a/configs/cerebras_1_3B_alpaca.yml +++ b/configs/cerebras_1_3B_alpaca.yml @@ -24,7 +24,7 @@ lora_fan_in_fan_out: false wandb_project: pythia-1.4b-lora wandb_watch: wandb_run_id: -wandb_log_model: checkpoint +wandb_log_model: output_dir: ./lora-alpaca batch_size: 32 micro_batch_size: 4