Skip to content

Commit 521a09d

Browse files
authored
Update get_chinese_llama.py
1 parent 8c50ee5 commit 521a09d

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

imagebind_LLM/tools/get_chinese_llama.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -33,7 +33,8 @@
3333
delta_dict = torch.load(os.path.join(delta_path, 'consolidated.00.pth'), map_location="cpu")
3434
new_dict = {}
3535
for key in ori_dict:
36+
if "rope" in key: continue
3637
new_value = (ori_dict[key].float() + delta_dict[key].float()).half()
3738
new_dict[key] = new_value
3839

39-
torch.save(new_dict, os.path.join(new_path, 'consolidated.00.pth'))
40+
torch.save(new_dict, os.path.join(new_path, 'consolidated.00.pth'))

0 commit comments

Comments
 (0)