Fix for pytorch version check issue in __init__.py

[RIP-Skillan]

This is the solution as mentioned in the discussion #2660
It fixes for Pytorch version check issue for adding 'weights_only' argument. Added a more accurate version parser.

Original Code:

kwargs = {"weights_only": True} if torch.__version__ >= "1.13" else {}

Fixed Version Check using Version Parser:

from packaging import version
kwargs = {"weights_only": True} if version.parse(torch.__version__ ) >= version.parse("1.13") else {}