diff --git a/examples/inference/llava1_5.yaml b/examples/inference/llava1_5.yaml new file mode 100644 index 00000000..75035141 --- /dev/null +++ b/examples/inference/llava1_5.yaml @@ -0,0 +1,3 @@ +model_name_or_path: llava-hf/llava-1.5-7b-hf +template: vicuna +visual_inputs: true \ No newline at end of file diff --git a/examples/train_lora/llama3_lora_pretrain.yaml b/examples/train_lora/llama3_lora_pretrain.yaml index 5e8aaaef..7e3ea06c 100644 --- a/examples/train_lora/llama3_lora_pretrain.yaml +++ b/examples/train_lora/llama3_lora_pretrain.yaml @@ -15,7 +15,7 @@ overwrite_cache: true preprocessing_num_workers: 16 ### output -output_dir: saves/llama3-8b/lora/sft +output_dir: saves/llama3-8b/lora/pretrain logging_steps: 10 save_steps: 500 plot_loss: true