runtime error

Exit code: 1. Reason: peline File "/usr/local/lib/python3.10/site-packages/transformers/pipelines/audio_classification.py", line 21, in <module> from .base import Pipeline, build_pipeline_init_args File "/usr/local/lib/python3.10/site-packages/transformers/pipelines/base.py", line 34, in <module> from ..modelcard import ModelCard File "/usr/local/lib/python3.10/site-packages/transformers/modelcard.py", line 47, in <module> from .training_args import ParallelMode File "/usr/local/lib/python3.10/site-packages/transformers/training_args.py", line 75, in <module> from accelerate.state import AcceleratorState, PartialState File "/usr/local/lib/python3.10/site-packages/accelerate/__init__.py", line 16, in <module> from .accelerator import Accelerator File "/usr/local/lib/python3.10/site-packages/accelerate/accelerator.py", line 36, in <module> from accelerate.utils.dataclasses import FP8BackendType File "/usr/local/lib/python3.10/site-packages/accelerate/utils/__init__.py", line 14, in <module> from ..parallelism_config import ParallelismConfig File "/usr/local/lib/python3.10/site-packages/accelerate/parallelism_config.py", line 20, in <module> from torch.distributed.device_mesh import init_device_mesh ModuleNotFoundError: No module named 'torch.distributed.device_mesh' The above exception was the direct cause of the following exception: Traceback (most recent call last): File "/home/user/app/app.py", line 9, in <module> from transformers import pipeline File "<frozen importlib._bootstrap>", line 1075, in _handle_fromlist File "/usr/local/lib/python3.10/site-packages/transformers/utils/import_utils.py", line 1576, in __getattr__ module = self._get_module(self._class_to_module[name]) File "/usr/local/lib/python3.10/site-packages/transformers/utils/import_utils.py", line 1588, in _get_module raise RuntimeError( RuntimeError: Failed to import transformers.pipelines because of the following error (look up to see its traceback): No module named 'torch.distributed.device_mesh'

Container logs: