import torch weight = torch.load('defendant_model/adapter_model.bin', 'cpu') for k, v in weight.items(): print(k, v.shape)