Fix PyTorch 2.6 weights_only loading and syntax warnings

Fix torchvision import for compatibility with newer PyTorch versions

BIT