Merge pull request #4804 from codemayq/fix-examples

tiny fix of examples
2024-07-13 20:49:13 +08:00 · 2024-07-13 20:49:13 +08:00 · 32699a82a6
parent 67040f149c f618b80fa2
commit 32699a82a6
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