1. fix output_dir in llama3_lora_pretrain.yaml

2. add llava1_5.yaml for inference
2024-07-13 13:16:22 +08:00 · 2024-07-13 13:16:22 +08:00 · 982a1cdd24
parent 67040f149c
commit 982a1cdd24
2 changed files with 4 additions and 1 deletions
--- a/examples/inference/llava1_5.yaml
+++ 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
--- 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