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")