RLHF-Reward-Modeling icon indicating copy to clipboard operation
RLHF-Reward-Modeling copied to clipboard

Update eval_bench_mark.py allow use bf16 or f32

Open ZizhengYang opened this issue 1 year ago • 0 comments

Some environment does support bfloat16 that good, so adding a new argument, works similarly to bf16 parameter in gemma_rm.py

ZizhengYang avatar Apr 22 '24 12:04 ZizhengYang