ModuleNotFoundError: No module named 'pytorch_lightning.utilities.distributed'

[nice1999my]

I am getting error ModuleNotFoundError: No module named 'pytorch_lightning.utilities.distributed'

Please advise.

Traceback (most recent call last):
File "/home/studio-lab-user/sagemaker-studiolab-notebooks/stable-diffusion-webui/launch.py", line 39, in
main()
File "/home/studio-lab-user/sagemaker-studiolab-notebooks/stable-diffusion-webui/launch.py", line 35, in main
start()
File "/home/studio-lab-user/sagemaker-studiolab-notebooks/stable-diffusion-webui/modules/launch_utils.py", line 390, in start
import webui
File "/home/studio-lab-user/sagemaker-studiolab-notebooks/stable-diffusion-webui/webui.py", line 54, in
from modules.call_queue import wrap_gradio_gpu_call, wrap_queued_call, queue_lock # noqa: F401
File "/home/studio-lab-user/sagemaker-studiolab-notebooks/stable-diffusion-webui/modules/call_queue.py", line 6, in
from modules import shared, progress, errors
File "/home/studio-lab-user/sagemaker-studiolab-notebooks/stable-diffusion-webui/modules/shared.py", line 21, in
from ldm.models.diffusion.ddpm import LatentDiffusion
File "/home/studio-lab-user/sagemaker-studiolab-notebooks/stable-diffusion-webui/repositories/stable-diffusion-stability-ai/ldm/models/diffusion/ddpm.py", line 20, in
from pytorch_lightning.utilities.distributed import rank_zero_only
ModuleNotFoundError: No module named 'pytorch_lightning.utilities.distributed'

[daniloedu]

It seems that the notebook is not compatible anymore with SagaMaker StudioLab.

[03Sys]

Try executing these commands in a cell:

!pip install pytorch-lightning==1.6.5
!pip install lightning-utilities==0.4.0