Update functionalities to work with pytorch 2.6

[jrzaurin]

some of the internal functionalities to save the model will fail when using pytotch 2.6

E                           _pickle.UnpicklingError: Weights only load failed. This file can still be loaded, to do so you have two options, do those steps only if you trust the source of the checkpoint.
E                           	(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.
E                           	(2) Alternatively, to load with `weights_only=True` please check the recommended steps in the following error message.
E                           	WeightsUnpickler error: Unsupported global: GLOBAL pytorch_widedeep.models.wide_deep.WideDeep was not an allowed global by default. Please use `torch.serialization.add_safe_globals([WideDeep])` or the `torch.serialization.safe_globals([WideDeep])` context manager to allowlist this global if you trust this class/function.
E
E                           Check the documentation of torch.load to learn more about types accepted by default with weights_only https://pytorch.org/docs/stable/generated/torch.load.html.

Needs fixing

[jrzaurin]

For now see #241