update examples

2025-12-15 03:10:35 +08:00 · 2024-03-09 02:30:37 +08:00
parent e8dd38b7fd
commit 4c00bcdcae
6 changed files with 5 additions and 2 deletions
--- a/examples/extras/llama_pro/sft.sh
+++ b/examples/extras/llama_pro/sft.sh
@@ -10,6 +10,7 @@ CUDA_VISIBLE_DEVICES=0 python ../../../src/train_bash.py \
    --finetuning_type freeze \
    --name_module_trainable all \
    --num_layer_trainable 8 \
+    --use_llama_pro \
    --output_dir ../../../saves/LLaMA2-7B-Pro/lora/sft \
    --overwrite_cache \
    --overwrite_output_dir \