runtime error

Using cuda for inference. Traceback (most recent call last): File "/home/user/app/persona.py", line 12, in <module> from improve import improve, vid2frames, restore_frames File "/home/user/app/improve.py", line 7, in <module> from basicsr.utils import imwrite File "/home/user/.pyenv/versions/3.10.14/lib/python3.10/site-packages/basicsr/__init__.py", line 4, in <module> from .data import * File "/home/user/.pyenv/versions/3.10.14/lib/python3.10/site-packages/basicsr/data/__init__.py", line 22, in <module> _dataset_modules = [importlib.import_module(f'basicsr.data.{file_name}') for file_name in dataset_filenames] File "/home/user/.pyenv/versions/3.10.14/lib/python3.10/site-packages/basicsr/data/__init__.py", line 22, in <listcomp> _dataset_modules = [importlib.import_module(f'basicsr.data.{file_name}') for file_name in dataset_filenames] File "/home/user/.pyenv/versions/3.10.14/lib/python3.10/importlib/__init__.py", line 126, in import_module return _bootstrap._gcd_import(name[level:], package, level) File "/home/user/.pyenv/versions/3.10.14/lib/python3.10/site-packages/basicsr/data/realesrgan_dataset.py", line 11, in <module> from basicsr.data.degradations import circular_lowpass_kernel, random_mixed_kernels File "/home/user/.pyenv/versions/3.10.14/lib/python3.10/site-packages/basicsr/data/degradations.py", line 8, in <module> from torchvision.transforms.functional_tensor import rgb_to_grayscale ModuleNotFoundError: No module named 'torchvision.transforms.functional_tensor'

Container logs: