runtime error

smmap, setproctitle, sentry-sdk, regex, psutil, omegaconf, nvidia-nvtx-cu11, nvidia-nccl-cu11, nvidia-cusparse-cu11, nvidia-curand-cu11, nvidia-cufft-cu11, nvidia-cuda-runtime-cu11, nvidia-cuda-nvrtc-cu11, nvidia-cuda-cupti-cu11, nvidia-cublas-cu11, networkx, lightning-utilities, ftfy, einops, docker-pycreds, nvidia-cusolver-cu11, nvidia-cudnn-cu11, gitdb, GitPython, wandb, triton, torch, torchvision, torchmetrics, pytorch-lightning, clip Successfully installed GitPython-3.1.31 antlr4-python3-runtime-4.9.3 appdirs-1.4.4 clip-1.0 cmake-3.26.4 docker-pycreds-0.4.0 einops-0.6.1 ftfy-6.1.1 gitdb-4.0.10 lightning-utilities-0.8.0 lit-16.0.5.post0 mpmath-1.3.0 networkx-3.1 nvidia-cublas-cu11-11.10.3.66 nvidia-cuda-cupti-cu11-11.7.101 nvidia-cuda-nvrtc-cu11-11.7.99 nvidia-cuda-runtime-cu11-11.7.99 nvidia-cudnn-cu11-8.5.0.96 nvidia-cufft-cu11-10.9.0.58 nvidia-curand-cu11-10.2.10.91 nvidia-cusolver-cu11-11.4.0.1 nvidia-cusparse-cu11-11.7.4.91 nvidia-nccl-cu11-2.14.3 nvidia-nvtx-cu11-11.7.91 omegaconf-2.3.0 pathtools-0.1.2 psutil-5.9.5 pytorch-lightning-2.0.3 regex-2023.6.3 sentry-sdk-1.25.1 setproctitle-1.3.2 smmap-5.0.0 sympy-1.12 torch-2.0.1 torchmetrics-0.11.4 torchvision-0.15.2 triton-2.0.0 wandb-0.15.4 wcwidth-0.2.6 [notice] A new release of pip available: 22.3.1 -> 23.1.2 [notice] To update, run: python -m pip install --upgrade pip Traceback (most recent call last): File "/home/user/app/app.py", line 27, in <module> import train_latent_diffusion as train File "/home/user/app/./cloob-latent-diffusion/train_latent_diffusion.py", line 17, in <module> from taming.models import cond_transformer, vqgan File "/home/user/app/./cloob-latent-diffusion/taming-transformers/taming/models/cond_transformer.py", line 6, in <module> from main import instantiate_from_config File "/home/user/app/./cloob-latent-diffusion/latent-diffusion/main.py", line 17, in <module> from pytorch_lightning.utilities.distributed import rank_zero_only ModuleNotFoundError: No module named 'pytorch_lightning.utilities.distributed'

Container logs: