torchtune icon indicating copy to clipboard operation
torchtune copied to clipboard

safe_torch_load failed when resume from checkpoint

Open ScottHoang opened this issue 7 months ago • 6 comments

Location: torchtune/utils/_checkpointing/_checkpointer.py line: 438

Error: _pickle.UnpicklingError: Weights only load failed. Re-running torch.load with weights_only set to False will likely succeed, but it can result in arbitrary code execution.Do it only if you get the file from a trusted source. WeightsUnpickler error: Unsupported class omegaconf.listconfig.ListConfig

suggest fix: set weights_only to False

ScottHoang avatar Jul 03 '24 18:07 ScottHoang