change: change predict yaml

This commit is contained in:
wql 2024-08-15 11:18:12 +08:00
parent c97367ad0a
commit 0e8b03b638
1 changed files with 1 additions and 3 deletions

View File

@ -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