fix FutureWarning with torch.load

[HiDolen]

This pull request addresses the FutureWarning issue described in issue #26.

• Updated torch.load to use weights_only=True to avoid potential security risks and to comply with future PyTorch updates.
• Tested with PyTorch version 2.4.0.

[lucidrains]

@HiDolen how far back in torch versions does this work? should i enforce a minimum torch version?

[HiDolen]

@HiDolen how far back in torch versions does this work? should i enforce a minimum torch version?

The weights_only feature was added in version 1.13 of PyTorch. A FutureWarning will occur in version 2.4.

I think it's a good idea to have the version higher than 2.0.