Skip to content

Commit c1465d7

Browse files
committed
fix(pt): update type embedding dictionary handling in translate_type_embd_dicts_to_dpa2
1 parent cc0be59 commit c1465d7

1 file changed

Lines changed: 1 addition & 0 deletions

File tree

source/tests/pt/model/test_descriptor_dpa2.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -196,5 +196,6 @@ def translate_type_embd_dicts_to_dpa2(
196196
target_dict[tk] = type_embd_dict[kk]
197197
record[all_keys.index("repinit.compress_data.0")] = True
198198
record[all_keys.index("repinit.compress_info.0")] = True
199+
record[all_keys.index("repinit.type_embd_data")] = True
199200
assert all(record)
200201
return target_dict

0 commit comments

Comments
 (0)