Skip to content

Commit 2d83458

Browse files
committed
update reranker trainer
1 parent 58fecb4 commit 2d83458

2 files changed

Lines changed: 2 additions & 2 deletions

File tree

examples/finetune/embedder/README.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -294,7 +294,7 @@ Here are some new arguments:
294294

295295
```shell
296296
torchrun --nproc_per_node 2 \
297-
-m FlagEmbedding.finetune.embedder.decoder_only.base \
297+
-m FlagEmbedding.finetune.embedder.decoder_only.icl \
298298
--model_name_or_path BAAI/bge-en-icl \
299299
--cache_dir ./cache/model \
300300
--use_lora True \

examples/finetune/reranker/README.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -176,7 +176,7 @@ torchrun --nproc_per_node 2 \
176176
--use_flash_attn True \
177177
--target_modules q_proj k_proj v_proj o_proj \
178178
--save_merged_lora_model True \
179-
--model_type decoder
179+
--model_type decoder \
180180
--cache_dir ./cache/model \
181181
--train_data ./example_data/prompt_based/examples.jsonl \
182182
--cache_path ./cache/data \

0 commit comments

Comments
 (0)