Function load

tools/transformers/convert2hf.py:134–137

(fp)

    args = parser.parse_args()

    def load(fp):
        with open(fp, "rb") as f:
            pt_data = torch.load(f, map_location="cpu")
        return pt_data

    folder = args.src_folder
    target_folder = args.tgt_folder