incompatibility with the latest version of lightning utilities

[SteelMarley]

Hi! Just a heads-up — the Colab version of the AUTOMATIC1111 notebook is currently broken due to an incompatibility with the latest version of lightning_utilities.

The error is caused by a missing ModuleAvailableCache in lightning_utilities.core.imports — this function seems to have been removed in newer versions. Downgrading to lightning-utilities==0.9.0 fixes the issue.

Everything was working fine until yesterday. Maybe consider pinning the package version in the notebook to avoid future breakage?

Thanks for the great work 🙏

Hey everyone,

For those using TheLastBen’s Colab notebook for Stable Diffusion (AUTOMATIC1111) and suddenly getting import errors like:

ImportError: cannot import name 'ModuleAvailableCache' from 'lightning_utilities.core.imports'
Or errors involving pytorch_lightning, wandb, etc...

It’s because some core dependencies auto-updated and broke compatibility.

✅ TEMP FIX (Colab):
Run this before starting webui.py:

!pip uninstall -y lightning-utilities pytorch-lightning torchmetrics wandb
!pip install lightning-utilities==0.9.0
!pip install pytorch_lightning==1.9.4
!pip install torchmetrics==0.11.4
!pip install wandb==0.15.12

Also, if needed, patch these 2 files to avoid annoying warnings:

!sed -i 's@from pytorch_lightning.loggers.wandb import WandbLogger # noqa: F401@@g' /usr/local/lib/python3.11/dist-packages/pytorch_lightning/loggers/init.py
!sed -i 's@from .mailbox import ContextCancelledError@@g' /usr/local/lib/python3.11/dist-packages/wandb/sdk/lib/retry.py
!sed -i 's@raise ContextCancelledError("retry timeout")@print("retry timeout")@g' /usr/local/lib/python3.11/dist-packages/wandb/sdk/lib/retry.py
This brought everything back to normal for me.
Hope it helps while we wait for a clean fix in the repo. 🙏

Cheers!