Weight Issue with Pytorch 2.6 Bug Fx

[Daniel-moen]

Sitting on Pytorch 2.6 getting weight (1) In PyTorch 2.6, we changed the default value of the `weights_only` argument in `torch.load` from `False` to `True`. 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 got the file from a trusted source. (2) Alternatively, to load with `weights_only=True` please check the recommended steps in the following error message. WeightsUnpickler error: Unsupported global: GLOBAL numpy.core.multiarray.scalar was not an allowed global by default. Please use `torch.serialization.add_safe_globals([scalar])` or the `torch.serialization.safe_globals([scalar])` context manager to allowlist this global if you trust this class/function.
on Line 212
checkpoint = torch.load(ckpt_path, map_location=device)
Change it to:
checkpoint = torch.load(ckpt_path, map_location=device, weights_only=False)
This allows PyTorch to load the full model

[Daniel-moen]

Fixed by #619