diff --git a/configs/llama_7B_alpaca.yml b/configs/llama_7B_alpaca.yml index 20efd58d3..66af807dd 100644 --- a/configs/llama_7B_alpaca.yml +++ b/configs/llama_7B_alpaca.yml @@ -26,7 +26,7 @@ lora_fan_in_fan_out: false wandb_project: llama-7b-lora wandb_watch: wandb_run_id: -wandb_log_model: checkpoint +wandb_log_model: output_dir: ./lora-llama-alpaca batch_size: 128 micro_batch_size: 16