import torch | |
org_state_dict = torch.load("/home/cv/sxp/llava_new/llava/qwen_clip/pytorch_model.bin",map_location="cuda:1") | |
new_state_dict = {} | |
for key,value in org_state_dict.items(): | |
if "attn_pool" not in key: | |
new_state_dict[key] = value | |
print(new_state_dict.keys()) | |
torch.save(new_state_dict,"/home/cv/sxp/llava_new/llava/qwen_clip/pytorch_model_new.bin") |