github
/
InternLM
mirror of https://github.com/InternLM/InternLM
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

update

pull/751/head
zhulin1 2024-07-01 19:51:40 +08:00
parent d4d54639c6
commit 47338376a5
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
tests/test_hf_model.py
View File

@ -233,7 +233,8 @@ class TestReward:
rank_res = model.rank(tokenizer, [chat_1, chat_2]) rank_res = model.rank(tokenizer, [chat_1, chat_2])
print('rank_res: ', rank_res) # lower index means higher score print('rank_res: ', rank_res) # lower index means higher score
# >>> rank_res: [0, 1] # >>> rank_res: [0, 1]
assert rank_res[0] == 0 & rank_res[1] == 1 assert rank_res[0] == 0
assert rank_res[1] == 1
@pytest.mark.parametrize( @pytest.mark.parametrize(
'model_name', 'model_name',