mirror of
https://github.com/open-compass/opencompass.git
synced 2025-05-30 16:03:24 +08:00
[Fix] fix duplicate error in partitioner (#1552)
* fix pip version * fix pip version * fix duplicate error in paritioner * fix duplicate error in paritioner
This commit is contained in:
parent
335667183a
commit
a2e9bc0c41
@ -18,6 +18,8 @@ def remove_duplicate_pairs(model_combinations):
|
|||||||
combo_dict = {}
|
combo_dict = {}
|
||||||
for i, combo in enumerate(model_combinations):
|
for i, combo in enumerate(model_combinations):
|
||||||
sorted_names = tuple(sorted((combo[0]['abbr'], combo[1]['abbr'])))
|
sorted_names = tuple(sorted((combo[0]['abbr'], combo[1]['abbr'])))
|
||||||
|
if sorted_names[0] == sorted_names[1]:
|
||||||
|
continue
|
||||||
if sorted_names not in combo_dict:
|
if sorted_names not in combo_dict:
|
||||||
combo_dict[sorted_names] = i
|
combo_dict[sorted_names] = i
|
||||||
new_model_combinations = [
|
new_model_combinations = [
|
||||||
|
Loading…
Reference in New Issue
Block a user