diff --git a/configs/llama_7B_4bit.yml b/configs/llama_7B_4bit.yml index 0033a6e3a..feb4e21a1 100644 --- a/configs/llama_7B_4bit.yml +++ b/configs/llama_7B_4bit.yml @@ -24,7 +24,7 @@ lora_fan_in_fan_out: false wandb_project: wandb_watch: wandb_run_id: -wandb_log_model: checkpoint +wandb_log_model: output_dir: ./lora-test batch_size: 8 micro_batch_size: 2