change: change predict yaml

2024-08-15 11:18:12 +08:00 · 2024-08-15 11:18:12 +08:00 · 0e8b03b638
parent c97367ad0a
commit 0e8b03b638
1 changed files with 1 additions and 3 deletions
--- a/results/inference/Llama2-7B/llama2_lora_predict.yaml
+++ b/results/inference/Llama2-7B/llama2_lora_predict.yaml
@ -2,9 +2,7 @@
 model_name_or_path: modelscope/Llama-2-7b-ms

 ### method
-stage: sft
 do_predict: true
-finetuning_type: lora

 ### dataset
 eval_dataset: alpaca_gpt4_zh
@ -15,7 +13,7 @@ overwrite_cache: true
 preprocessing_num_workers: 16

 ### output
-output_dir: ./results/inference/Llama2-7B/Llama2-7B_inference_08_15_11_01
+output_dir: ./results/inference/Llama2-7B/Llama2-7B_inference_08_15_11_06
 overwrite_output_dir: true

 ### eval