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 17:30:17 +08:00
parent 7df447a572
commit d474c10dae
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
tests/test_hf_model.py
View File

@ -219,8 +219,8 @@ class TestReward:
# batch inference, get multiple scores at once # batch inference, get multiple scores at once
scores = model.get_scores(tokenizer, [chat_1, chat_2]) scores = model.get_scores(tokenizer, [chat_1, chat_2])
print('scores: ', scores) print('scores: ', scores)
assert scores[0] > 0 assert scores[0][0] > 0
assert scores[1] < 0 assert scores[1][0] < 0
# compare whether chat_1 is better than chat_2 # compare whether chat_1 is better than chat_2
compare_res = model.compare(tokenizer, chat_1, chat_2) compare_res = model.compare(tokenizer, chat_1, chat_2)
@ -313,6 +313,7 @@ class TestReward:
# print the best response # print the best response
best_response = sorted_candidates[0][1][-1]['content'] best_response = sorted_candidates[0][1][-1]['content']
print(sorted_candidates)
print(best_response) print(best_response)
assert len(sorted_candidates) == 3 assert len(sorted_candidates) == 3