Skip to content

Commit

Permalink
[Usability] better to set overwrite_cache for rm inference in online …
Browse files Browse the repository at this point in the history
…rlhf
  • Loading branch information
wheresmyhair committed Jun 23, 2024
1 parent c9c93de commit 9f0a1da
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions scripts/run_rm_inference.sh
Original file line number Diff line number Diff line change
Expand Up @@ -65,6 +65,7 @@ accelerate launch --config_file configs/accelerator_multigpu_config.yaml \
--block_size 4096 \
--inference_batch_size 16 \
--dataset_path ${dataset_path} \
--overwrite_cache True \
--conversation_template ${conversation_template} \
--preprocessing_num_workers 16 \
--save_results True \
Expand Down

0 comments on commit 9f0a1da

Please sign in to comment.