runtime error

Traceback (most recent call last): File "MED_VQA_Huggyface_Gradio.py", line 4, in <module> from transformers import ViltProcessor, ViltForQuestionAnswering File "/home/user/.local/lib/python3.8/site-packages/transformers/__init__.py", line 26, in <module> from . import dependency_versions_check File "/home/user/.local/lib/python3.8/site-packages/transformers/dependency_versions_check.py", line 16, in <module> from .utils.versions import require_version, require_version_core File "/home/user/.local/lib/python3.8/site-packages/transformers/utils/__init__.py", line 31, in <module> from .generic import ( File "/home/user/.local/lib/python3.8/site-packages/transformers/utils/generic.py", line 432, in <module> import torch.utils._pytree as _torch_pytree ModuleNotFoundError: No module named 'torch.utils._pytree'

Container logs: