state_dict = torch.load(model_path, weights_only=True, map_location=lambda storage, loc: storage) f24abb2 verified Fabrice-TIERCELIN commited on Feb 9