ImportError: cannot import name 'model_urls' from 'torchvision.models.resnet'

[ZhilinGuo]

I have successfully installed everything in Installation instructions. However, I'm getting this error when running the demo code.

ImportError: cannot import name 'model_urls' from 'torchvision.models.resnet' (/home/user/.conda/envs/shapy/lib/python3.8/site-packages/torchvision/models/resnet.py)

I understand that the original repo was tested with Python 3.8, CUDA 10.2 and PyTorch 1.7.1. But due to a constraint of my GPU I can only use CUDA 11.X +. Thus I'm current using CUDA 11.8 and PyTorch 2.1.1.

Am I out of luck trying to run this code on my setup?